Skip to content

Commit

Permalink
wip update readme
Browse files Browse the repository at this point in the history
  • Loading branch information
akelad committed Nov 26, 2018
1 parent 355ece6 commit 0e54f35
Show file tree
Hide file tree
Showing 4 changed files with 38 additions and 3 deletions.
17 changes: 14 additions & 3 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,20 @@ help:
@echo " run"
@echo " Runs the bot on the commandline"

train-core:
# TODO: fix
python -m rasa_core.train -s data/core/ -d domain.yml -o models/dialogue --epochs 300
train-redp:
python -m rasa_core.train -s data/core/ -d domain.yml -o models/dialogue -c redp.yml --augmentation_factor 0

train-lstm-bin:
python -m rasa_core.train -s data/core/ -d domain.yml -o models/dialogue -c lstm_bin.yml --augmentation_factor 0

train-lstm-feat:
python -m rasa_core.train -s data/core/ -d domain.yml -o models/dialogue -c lstm_feat.yml --augmentation_factor 0

train-compare:
python -m rasa_core.train compare --stories data/core/ -d domain.yml -o comparison_models -c redp.yml lstm_bin.yml lstm_feat.yml --augmentation_factor 0 --runs 5

evaluate-compare:
python -m rasa_core.evaluate compare --stories data/core/ --core comparison_models -o results/

run-core:
python -m rasa_core.run --core models/dialogue --endpoints endpoints.yml
Expand Down
10 changes: 10 additions & 0 deletions lstm_bin.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
policy:
- name: KerasPolicy
batch_size: 32
epochs: 400
rnn_size: 64
featurizer:
- name: MaxHistoryTrackerFeaturizer
max_history: 38
state_featurizer:
- name: LabelTokenizerSingleStateFeaturizer
10 changes: 10 additions & 0 deletions lstm_feat.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
policy:
- name: KerasPolicy
batch_size: 32
epochs: 400
rnn_size: 64
featurizer:
- name: MaxHistoryTrackerFeaturizer
max_history: 38
state_featurizer:
- name: BinarySingleStateFeaturizer
4 changes: 4 additions & 0 deletions redp.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
policy:
- name: EmbeddingPolicy
epochs: 2000
attn_shift_range: 5

0 comments on commit 0e54f35

Please sign in to comment.