a simple message reply suggestion system
Have you ever seen or used Google Smart Reply? It's a service that provides automatic reply suggestions for user messages. See below.
This is a useful application of the retrieval based chatbot. Think about it. How many times do we text a message like thx, hey, or see you later? In this project, we build a simple message reply suggestion system.
Kyubyong Park
Code-review by Yj Choe
We fine-tune huggingface's the Bert pretrained model for sequence classification. In it, a special starting token [CLS] stores the entire information of a sentence. Extra layers are attached to project the condensed information to classification units (here 100).
bash download.sh
python construct_sg.py
python make_phr2sg_id.py
python encode.py
python prepro.py
python train.py
python test.py --ckpt log/9500_ACC0.1.pt