Skip to content

chatbot fine-tuned on pre-trained model with classification algorithm applied

Notifications You must be signed in to change notification settings

hyunmila/chatbot

Repository files navigation

chatbot

classification_training

Chatbot model trained using neural networks. The training model structure is an extended version of code found here: https://github.com/amilavm/Chatbot_Keras.

transformers_bot

Chatbot model fine-tuned DialoGPT with my own dataset of conversations using Huggingface Transformers library and Keras Tensorflow.