This project predicts house prices using the Ames Housing dataset. Multiple ensemble regression models are trained and evaluated, with feature correlation analysis for explainability.
- Ames Housing Dataset
- 2,900+ records
- 80+ features
- Dropped high-missing categorical columns
- Mean imputation for numerical features
- Mode imputation for categorical features
- One-hot encoding
- Feature scaling using StandardScaler
- Pearson correlation with target variable
- Top 20 correlated features selected
- Random Forest Regressor
- Gradient Boosting Regressor
- HistGradientBoosting Regressor
- R² Score
- RMSE (Root Mean Squared Error)
| Model | Test R² | RMSE |
|---|---|---|
| Random Forest | XX | XX |
| Gradient Boosting | XX | XX |
| HistGradientBoosting | XX | XX |
- Python
- Pandas, NumPy
- Scikit-learn
- Matplotlib, Seaborn
pip install -r requirements.txt
python src/train_models.py