automl / models /base_model.py

Commit History

Implement model explainability using SHAP, including global importance, summary plots, and single-row waterfalls. Add abstract base model class for model wrappers, create a model zoo for various estimators based on dataset size and task type, and develop a cross-validation trainer for model evaluation. Integrate MLflow for experiment tracking, logging dataset stats, model results, and SHAP plots. Introduce Optuna-based hyperparameter tuning for top models from the leaderboard, with parameter suggestion spaces for various algorithms.
e56bd39

Aju360 commited on

Add initial project structure and core modules for AutoML system
246c042

Aju360 commited on