iotarepeat/sms-spam-detector

A machine learning model created to fight sms spam uses Naive Bayes classifier

Python

Training

Run train.py to generate required files for prediction
It will print the accuracy, at time of writing accuracy was 97%

Usage

Run main.py and enter your message
It will print whether the message is spam/ham

Credits

SMSSpamCollection dataset - https://archive.ics.uci.edu/ml/datasets/sms+spam+collection
Stopwords - https://github.com/stopwords-iso/stopwords-en