This project predicts product sales based on advertising spend across different media channels (TV, Radio, Newspaper). It demonstrates a complete machine learning workflow from data exploration to model deployment.
A company wants to understand how their advertising budget across different media channels affects sales. Using historical data, we build a predictive model to forecast sales and optimize advertising spend allocation.
The dataset contains 200 observations with 4 variables:
- TV: Advertising spend on TV (in thousands of dollars)
- Radio: Advertising spend on Radio (in thousands of dollars)
- Newspaper: Advertising spend on Newspaper (in thousands of dollars)
- Sales: Product sales (in thousands of units)
- Python 3.8+
- Pandas & NumPy (Data manipulation)
- Matplotlib & Seaborn (Visualization)
- Scikit-learn (Machine Learning)
- Data exploration and visualization
- Correlation analysis
- Linear Regression model
- Model evaluation metrics (RΒ², RMSE)
- Sales prediction function
pip install -r requirements.txt