-
Idea
- Appeal that it is an idea that everyone would love. Everyone has a situation that whether or not they should send some risky message or not.
-
Technology used
- BERT(Bidirectional Transformer) Model
- Binary Sentiment Classification
-
Scalability
- Train and test with other dataset that has multiple sentiment
- Market demand
-
Future improvement
- Train with other dataset that has multiple sentiment
- Improve to implement a real chatbot
To get the model check point file, you can either
- Run Sentiment_Analysis/Transformers for Sentiment Analysis.ipynb (Recommended to run it on GPU environment)
- Download Bert-model.pt and place it in your working directory, or
path="./Bert-model.pt"
to fitmodel_loader
function.