A Deep Learning Text Classification model using Word Embeddings.
This is a deep learning based spam-ham classifier which is trained using keras Embedding layer.
Model architecture is as follows:
Model: "sequential"
Layer (type) |
---|
embedding (Embedding) |
flatten (Flatten) |
dense (Dense) |
To install requirements, do
$pip install requirements.txt
To train your own text classification model, do
$python train.py
To start webapp, do
$python app.py