This code implement and develop from CNN for Text Classification in Tensorflow & Classify text with Bert Tensorflow" blog post.Text classification with Bert
It is slightly simplified implementation of Kim's Convolutional Neural Networks for Sentence Classification paper in Tensorflow.
- Python 3
- Tensorflow==2.4.1
- Numpy
- Flask
With CNN train.py With Bert classify_text_with_bert.ipynb
Download model from https://drive.google.com/file/d/1mgIAVHMJTydz_9Omwm9pzqfNFwqMEf7J/view?usp=sharing
./eval.py --eval_train --checkpoint_dir="./runs/1459637919/checkpoints/"
Replace the checkpoint dir with the output from the training. To use your own data, change the eval.py
script to load your data.
API: sudo python3 bert_sv.py WEB U.I: npm start