๐ Overview
This project utilizes Linear Regression to predict the Total Price of a product based on multiple input features. It includes a Streamlit-based web application for user interaction.
๐ Features
- ML Model: Linear Regression trained using scikit-learn.
- Inputs (X):
beverage_categoryproduct_catcustomer_catregion_catUnit_PriceQuantityDiscount
- Output (Y):
Total_Price - Frontend: Built with Streamlit for real-time prediction.
๐ ๏ธ Tech Stack
- Python (Pandas, NumPy, Scikit-Learn)
- Streamlit (Web UI)
- Jupyter Notebook (for model training)
๐ How to Run
- Clone the Repository
git clone https://github.com/yourusername/your-repo-name.git cd your-repo-name - vscode terminal python -m streamlit run salary.py
Dataset - https://www.kaggle.com/api/v1/datasets/download/sebastianwillmann/beverage-sales
๐ Model Training The dataset is preprocessed and split into training and testing sets. The Linear Regression model is trained using sklearn.linear_model.LinearRegression. Model evaluation metrics include Mean Squared Error (MSE) and Rยฒ Score.
๐ Future Improvements Implement hyperparameter tuning for better accuracy. Support for multiple regression models. Deploy the model using Docker / Cloud platforms.
๐ Author Developed by Praveen R