This project demonstrates hyperparameter tuning on a regression model using Scikit-Learn's GridSearchCV, evaluated with Mean Squared Error (MSE), and tracked using MLflow for experiment management.
- Model Optimization via Grid Search over a defined hyperparameter space
- Cross-Validation with
cv=3to ensure model generalization - Metric Used: Mean Squared Error (MSE) for evaluation
- Experiment Tracking with MLflow (parameters, metrics, artifacts, and models)
- Best Model Selection based on the lowest average MSE across folds
- Python
- Scikit-Learn
- MLflow
- Pandas
βββ mlruns/ # MLflow logs (auto-generated)
βββ house_prediction.py # Script to run tuning + tracking
βββ requirements.txt # Required packages
βββ README.md # You're here!