- Create RandomForestModel and XGBoostModel wrappers with class weight support - Implement temporal and random train/val/test splitting - Add MLflow experiment tracking with full parameter and metric logging - Create evaluation module for confusion matrix, feature importance, and classification reports - Implement model training with sklearn/xgboost flavor logging and optional registry registration - Store training run metadata in PostgreSQL - Wire training stage into pipeline.py orchestrator - Support both RandomForest and XGBoost models with configurable hyperparameters |
||
|---|---|---|
| .. | ||
| .dvc | ||
| app | ||
| candle_ml.egg-info | ||
| config | ||
| features | ||
| migrations | ||
| training | ||
| .dvcignore | ||
| Dockerfile | ||
| pipeline.py | ||
| pyproject.toml | ||