ZoziLaMalice / Spam_Detector

Spam Detector with NLP model & Flask app

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Spam Detector

Project Description

The objective of this project was to develop a Sklearn NLP model and deploy it with the Flask framework.

I used the Multinomial Naives Bayes model with GridSearch tuning, as well as the Sklearn pipelines for data processing.

The F1 score of this model is very good on the test set: 0.95

Since the dataset was quite small, I chose to train the model online, that is to say each time the user wants to make a prediction, the model must train. It would have been better to train the model offline, then dump it in pickle format and then use it in online prediction.

Special thanks to Susan Li -> GitHub & the article.

About

Spam Detector with NLP model & Flask app


Languages

Language:Jupyter Notebook 93.3%Language:HTML 4.1%Language:CSS 1.9%Language:Python 0.7%