ποΈ Compressive Strength & π± COβ Emission Prediction Smart Material Property Prediction using Machine Learning
π Overview In this project, we predict the compressive strength of materials and estimate the COβ emissions during production using CatBoost with Optuna-based hyperparameter optimization.
Why?
Material engineers need accurate strength predictions to ensure safety and efficiency.
Environmental researchers require COβ emission estimation for sustainable production.
Our model aims to help manufacturers, researchers, and decision-makers make data-driven and eco-conscious choices.
π Features
β Predicts Compressive Strength of materials
β Estimates COβ emissions during production
β Uses CatBoost for superior accuracy
β Optuna for hyperparameter optimization
β Clean & reproducible code
β Easy-to-use for both research and industry applications
π§ Methodology
1οΈβ£ Dataset Preprocessing
- Missing value handling
- Outlier detection & removal
- Feature scaling & encoding
2οΈβ£ Model Selection
- Chose CatBoost for its handling of categorical data, robustness, and interpretability
3οΈβ£ Model Training
- Separate models for Compressive Strength and COβ Emission prediction
- Train-test split & cross-validation
4οΈβ£ Model Optimization (Optuna)
- Hyperparameters like depth, learning_rate, iterations tuned for best performance
- Objective function maximizes model accuracy while minimizing RMSE
π Results
Compressive Strength Model β High RΒ² score (>0.9)
COβ Emission Model β Accurate estimation within low error margins
Optimized models outperform default CatBoost settings
π Project Structure
π¦ compressive-strength-co2-prediction
β£ π data/ # Dataset files
β£ π notebooks/ # Jupyter notebooks for analysis
β£ π src/ # Main source code
β β£ model_training.py # Model training logic
β β£ optimize_model.py # Optuna optimization
β£ π requirements.txt # Dependencies
β£ π README.md # Project README
π οΈ Tech Stack
Python π
CatBoost π
Optuna βοΈ
Pandas, NumPy, Matplotlib, Seaborn
π¦ Installation & Usage
git clone https://github.com/yourusername/compressive-strength-co2-prediction.git
comparison-4 models.ipynb
CatBoost optimization.ipynb
π Future Improvements
πΉ Add SHAP explainability for feature importance
πΉ Include real-time prediction API
πΉ Expand dataset for global applicability
π€ Contributing
Pull requests are welcome! If you have ideas to make this better, please fork the repo and submit a PR.