This project predicts product sales based on advertising budgets using Machine Learning.
The dataset contains marketing spend on TV, Radio, and Newspaper and the corresponding Sales.
The goal is to identify which advertising channel impacts sales the most and build a predictive model.
βββ sales_prediction_project.py # Main ML script
βββ Sales_Prediction_Project.pptx # Project Presentation
βββ advertising.csv (optional) # Dataset
βββ README.md # Project Documentation
- Python
- Pandas
- NumPy
- Matplotlib / Seaborn
- Scikit-learn
- Linear Regression
- TV has the strongest impact on Sales
- Radio has moderate contribution
- Newspaper has very low impact
- Correlation heatmap confirms strong linear relationship with TV
- Simple
- Fast
- Interpretable
- Works well for numeric prediction
| Metric | Value |
|---|---|
| RΒ² Score | ~0.90 |
| RMSE | ~1.25 |
| MAE | ~1.02 |
Example prediction from model:
predict_sales(tv=150, radio=20, newspaper=15)python sales_prediction_project.py
Make sure the dataset advertising.csv is in the same folder.
- Add Random Forest & XGBoost
- Hyperparameter tuning
- Add more marketing features
- Deploy model using Flask or Streamlit
Shreyash Hedaoo
Machine Learning & Data Science Learner