Skip to content

Simple chat bot with training pipeline and pytorch 2 layers feed forward NN

Notifications You must be signed in to change notification settings

janrth/chat_bot

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Basic chat bot for the purpose of holiday booking/information for hotels.

A json file contains the intents, which are used during training. Stemming and bag-of-words is applied to decode the sentences. A simple feed forward Neural Network using pytorch is utilized for training. The chat bot is activated via a python file. If the soft-max probability is below a threshold the chat bot will not return the learned answer, but will ask to rephrase the question.

To train the model use the terminal and type:

  • python train_chatbot.py

To start the trained bot for interaction use the terminal and type:

  • python chat.py

About

Simple chat bot with training pipeline and pytorch 2 layers feed forward NN

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages