This project leverages machine learning to predict sales outcomes based on advertising campaign data. Using Linear and Polynomial Regression models, it analyzes the relationships between various advertising inputs and resulting sales figures.
-
Data Loading and Preprocessing
- Loads a dataset (
SalesPrediction.csv) with advertising metrics. - Preprocesses data by encoding categorical variables (One-Hot Encoding for
Influencertype) and handling missing values. - Splits data into training and test sets (70-30 split).
- Loads a dataset (
-
Feature Engineering
- Standardizes features using
StandardScaler. - Generates polynomial features for non-linear modeling.
- Standardizes features using
-
Model Training and Evaluation
- Trains Linear Regression and Polynomial Regression models.
- Evaluates model performance using metrics like R-squared.
Ensure you have Python 3.6+ and the following libraries:
pip install pandas numpy scikit-learn- Run the Notebook: Execute each cell sequentially to preprocess data, train models, and view results.
- Customize Parameters: Modify model parameters like
degreefor Polynomial Regression to test different configurations.
The model outputs predicted sales values and evaluates performance, providing insights into the impact of different advertising features on sales.
This project is for educational purposes.