Skip to content

Latest commit

 

History

2 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 

Repository files navigation

📊 Day 02: Machine Learning Pipeline & Hyperparameter Tuning

Python XGBoost Optuna Scikit-Learn

An automated machine learning modeling pipeline that compares classical algorithms, tunes hyperparameters using Bayesian search with Optuna, and evaluates performance with classification metrics and saved model artifacts.


🛠️ Key Features

  • Model Comparison: Cross-validation across Logistic Regression, Random Forest, and XGBoost.
  • Bayesian Optimization: Automated hyperparameter tuning using Optuna to maximize F1-Score.
  • Evaluation Artifacts: Generates Confusion Matrix visual reports and computes ROC-AUC metrics.
  • Model Serialization: Exports the trained model to best_xgboost_model.joblib for production serving.

📂 Repository Structure

Day02_Machine_Learning_Pipeline/
├── artifacts/
│   ├── best_xgboost_model.joblib   # Serialized model
│   └── confusion_matrix.png        # Evaluation plot
├── ml_pipeline.py                  # Core ML training & tuning script
├── .gitignore
└── README.md

🚀 How to Run
1. Setup Environment
Bash
python -m venv venv
source venv/bin/activate  # On Windows: venv\Scripts\activate
pip install pandas numpy scikit-learn xgboost optuna matplotlib seaborn joblib
2. Execute Training & Tuning
Bash
python ml_pipeline.py
Part of the 7-Day Machine Learning Engineering Challenge.

About

End-to-end Machine Learning classification pipeline with XGBoost, automated Optuna hyperparameter optimization, and comprehensive evaluation metrics. Part of the 7-Day ML Engineering Challenge.

Topics

Resources

Stars

2 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages