/nn-spam-detector

A neural network e-mail spam classifier.

Primary LanguagePython

E-mail spam detection

A fully vectorized implementation of a neural network, used for email spam detection. The dataset can be found under the following link.

Installing dependencies

This program is written in Python 3.7. Please use pip package manager to install the necessary dependencies (it is recommended to install them in a virtual environment like venv).

Installation using pip:

cd [project_directory]
pip install -r requirements.txt

Running tests

To train and test the model please run the following command.

python run.py