While the current model performs well, there are several promising directions for future work:
Ridge or Lasso Regression
L1/L2 regularisation to reduce potential overfitting risk
Random Forest Regressor
Capture non-linear relationships for better accuracy
XGBoost / LightGBM
State-of-the-art boosting, likely +5–10% R² gain
Hyperparameter Tuning
GridSearchCV to optimise model configuration
Feature: Car Age
2025 - year is more intuitive than raw year
Feature: Mileage per Year
Captures intensity of vehicle use more precisely
SHAP Values
More rigorous feature attribution and explainability
Data Quality Cleaning
Remove outliers (year=2060, MPG>150) for cleaner training
Log-Transform Target
Reduce impact of right-skew, potentially improve RMSE
Separate Hybrid/Electric Model
Better pricing for premium fuel types with few examples
MLflow Experiment Tracking
Reproducible ML experiments at scale
FastAPI Prediction Endpoint
REST API for real-time inference
Streamlit Web Dashboard
Interactive demo for non-technical users
Docker Containerisation
Consistent deployment across environments
Cloud Deployment
AWS/Azure/GCP for production-ready scalable inference
CI/CD Pipeline
Automated testing and retraining on new data