Skip to content

Commit

Permalink
Add configuration files and data
Browse files Browse the repository at this point in the history
  • Loading branch information
NoB0 committed Dec 6, 2023
1 parent 1dad29e commit 30b1659
Show file tree
Hide file tree
Showing 11 changed files with 27,015 additions and 0 deletions.
29 changes: 29 additions & 0 deletions reinforcement_learning/config/moviebot_config_rb_nlu.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
---
CONVERSATION_LOGS: # implementation: save conversational logs for debugging
save: False
nlp: True
path: reports/conversations/

DATA:
domain_path: data/movies_domain.yaml
db_path: data/movies_dbase.db
slot_values_path: data/slot_values.json

NLU:
tag_words_slots: config/tag_words_slots.json

RECOMMENDER: "slot_based"

TELEGRAM: False # execute the code on Telegram

POLLING: False # True when using Telegram without server

FLASK: False

BOT_TOKEN_PATH: config/bot_token.yaml

BOT_HISTORY: # save the conversational logs for Telegram users
save: True
path: conversation_history/

DEBUG: False
35 changes: 35 additions & 0 deletions reinforcement_learning/config/usersim_agenda_config.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
simulator_type: "agenda_based"
simulator_id: "usersim_moviebot_neural_nlu"

agent_id: "IAI MovieBot"
output_name: "moviebot"
# By default, the agent has an HTTP API.
agent_uri: "http://127.0.0.1:5001"

domain: reinforcement_learning/data/simulator/movies_domain.yaml
intents: reinforcement_learning/data/simulator/moviebot_interaction_model.yaml

items: reinforcement_learning/data/movielens-25m-sample/movies_w_keywords.csv
id_col: movieId
domain_mapping:
title:
slot: TITLE
genres:
slot: GENRE
multi-valued: True
delimiter: "|"
keywords:
slot: KEYWORD
multi-valued: True
delimiter: "|"
ratings: reinforcement_learning/data/movielens-25m-sample/ratings.csv
historical_ratings_ratio: 0.8

dialogues: reinforcement_learning/data/simulator/moviebot-1.1/annotated_dialogues_moviebot_1_1_updated.json
intent_classifier: "diet"
# If using the DIET classifier the following file needs to be provided.
rasa_dialogues: reinforcement_learning/data/simulator/moviebot-1.1/annotated_dialogues_moviebot_1_1_updated_rasa_agent.yaml

nlg: conditional

debug: True
13,817 changes: 13,817 additions & 0 deletions reinforcement_learning/data/movielens-25m-sample/movies_w_keywords.csv

Large diffs are not rendered by default.

Loading

0 comments on commit 30b1659

Please sign in to comment.