-
Notifications
You must be signed in to change notification settings - Fork 58
/
config.yml
45 lines (42 loc) · 1.63 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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
# Configuration for Rasa NLU.
# https://rasa.com/docs/rasa/nlu/components/
language: zh
pipeline:
# # No configuration for the NLU pipeline was provided. The following default pipeline was used to train your model.
# # If you'd like to customize it, uncomment and adjust the pipeline.
# # See https://rasa.com/docs/rasa/tuning-your-model for more information.
# - name: compoments.nlu.helpers.correction.TextCorrection # 文本纠错 Component
- name: compoments.nlu.tokenizers.bert_tokenizer.CustomBertTokenizer
cache_dir: ./tmp
model_weights: pre_models
- name: compoments.nlu.featurizer.lm_featurizer.OnnxLanguageModelFeaturizer
cache_dir: ./tmp
model_name: bert
model_weights: pre_models
onnx: false # 因为有依赖,默认未开启
quantize: false
- name: DIETClassifier
epochs: 100
- name: RegexEntityExtractor
- name: EntitySynonymMapper
- name: ResponseSelector
epochs: 100
retrieval_intent: rasa_faq
- name: FallbackClassifier
threshold: 0.3
ambiguity_threshold: 0.1
# Configuration for Rasa Core.
# https://rasa.com/docs/rasa/core/policies/
policies:
# # No configuration for policies was provided. The following default policies were used to train your model.
# # If you'd like to customize them, uncomment and adjust the policies.
# # See https://rasa.com/docs/rasa/policies for more information.
- name: AugmentedMemoizationPolicy
- name: TEDPolicy
max_history: 5
epochs: 100
- name: compoments.polices.ask_again_policy.AskAgainRulePolicy
core_fallback_threshold: 0.5
ask_again_intent: "再问一次"
need_ask_again_intents:
- "查询天气"