This repository contains a Python script implementing Gaussian Naive Bayes classification to predict social media campaign purchases based on user demographics. The script also provides an interactive prediction for a new user based on user input.
This repository demonstrates the use of Gaussian Naive Bayes classification to predict social media campaign purchases based on user demographics. The script includes model training, evaluation, and an interactive prediction feature for new user input.
The dataset used for training is located in the Resources folder: whoBoughtSocialMediaCampaign.csv
. It includes features such as age, salary, and gender.
- Run the script:
GAUSSIAN ALGORITHIM.ipynb
- Enter the user's age, salary, and gender when prompted.
- The script will provide a prediction on whether the user is likely to make a purchase
- pandas
- scikit-learn
- seaborn