-
Notifications
You must be signed in to change notification settings - Fork 1
/
config.yml
29 lines (26 loc) · 1.19 KB
/
config.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
#
language: en
# taking text input, extracting numbers from it (transformations)
pipeline:
- name: WhitespaceTokenizer # break text into words (whitespaces)
- name: RegexFeaturizer # regular expressions
- name: LexicalSyntacticFeaturizer
- name: CountVectorsFeaturizer
- name: CountVectorsFeaturizer
analyzer: "char_wb"
min_ngram: 1
max_ngram: 4
- name: DIETClassifier # classifier that does intents (what is the user trying to say) and
# entities (things that the user is referring to that chatbot knows about)
epochs: 100
- name: EntitySynonymMapper
- name: ResponseSelector
epochs: 100
# dialog policies, determines course of the conversation, what do I say next
policies:
- name: MemoizationPolicy # if we have seen this in stories, follow that path
- name: RulePolicy # if one of the provided rules is triggered, do what the rule says (rules for form for example)
- name: TEDPolicy # transformer driven policy, neural network...
max_history: 5 # attention span of the assistant, bigger = longer to train, but might get a better result
epochs: 100 # rounds of training
# ^ these hyperparameters are specifically for the policy