-
Notifications
You must be signed in to change notification settings - Fork 0
kausik-l/RLChatbot
Folders and files
Name | Name | Last commit message | Last commit date | |
---|---|---|---|---|
Latest commit | ||||
Repository files navigation
Instructions : 1) Create a virtual environment (Either Conda or Pip) and install the requirements using requirements.txt by running the command, conda create --name <env> --file requirements.txt 2) To get all the genres that are available in the dataset use, python cor_reader.py --show_genres 3) To train the bot from scratch with Cornell movie dialogues dataset using cross-entropy method, python train_crossent.py --cuda --data <genre> -n <name_of_save_folder> or just use 'python train_crossent.py --help' to know about different arguments that can be given Example : python train_crossent.py --cuda --data family -n crossent-family 4) To train using SCST method, python train_scst.py --cuda --data <genre> -l <saved_crossentropy_model_file> -n <name_of_save_folder> Example : python train_scst.py --cuda --data family -l saves/xe-family/epoch_040_0.853_0.185.dat -n sc-family 5) To test the model, python use_model.py -m <saved_model_location> -s "<input_phrase>" To make the bot interact with itself, --self argument can be used. Example : python use_model.py -m saves/scst-horror/epoch_040_0.629_0.138.dat -s "hi" --self 10
About
No description, website, or topics provided.
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published