An open API service indexing awesome lists of open source software.

https://github.com/akashprak/socialnetworkads

Predicting customer purchase behavior from the Social Network Ads dataset.
https://github.com/akashprak/socialnetworkads

data-analysis machine-learning mlflow pandas python scikit-learn seaborn xgboost

Last synced: about 1 year ago
JSON representation

Predicting customer purchase behavior from the Social Network Ads dataset.

Awesome Lists containing this project

README

          

### Social Network Ads - Performance Prediction
## Project Overview
The project implements various __bagging__ and __boosting__ models to predict whether a user purchases a product after viewing its social network Ad.
MLflow is used to handle the machine learning workflow. Various options for the python command line interface is also provided such as selecting which model to train, specifying parameters etc.

## Dataset Information
### columns
- `Age` : Age of user
- `EstimatedSalary` : Salary of the user
- `Purchased` : Binary label
- 0 : No
- 1 : Yes

The dataset includes age and estimated salary of the user. The purchased column indicates weather the particular user with age and estimated salary have bought the product or not by viewing the social ads of the product .

Dataset: https://www.kaggle.com/datasets/shub99/social-network-ads

## Dependencies
- `pandas`
- `numpy`
- `seaborn`
- `matplotlib`
- `xgboost`
- `scikit-learn`

```
pip install pandas numpy seaborn matplotlib xgboost scikit-learn
```