|
|
|
@ -5,24 +5,24 @@ pipeline:
|
|
|
|
|
- name: JiebaTokenizer
|
|
|
|
|
- name: LanguageModelFeaturizer
|
|
|
|
|
model_name: bert
|
|
|
|
|
model_weights: bert-base-chinese
|
|
|
|
|
- name: RegexFeaturizer
|
|
|
|
|
- name: DIETClassifier
|
|
|
|
|
epochs: 100
|
|
|
|
|
learning_rate: 0.001
|
|
|
|
|
tensorboard_log_directory: ./log
|
|
|
|
|
- name: ResponseSelector
|
|
|
|
|
epochs: 100
|
|
|
|
|
learning_rate: 0.001
|
|
|
|
|
- name: FallbackClassifier
|
|
|
|
|
threshold: 0.87
|
|
|
|
|
ambiguity_threshold: 0.1
|
|
|
|
|
- name: EntitySynonymMapper
|
|
|
|
|
model_weights: bert-base-chinese
|
|
|
|
|
- name: RegexFeaturizer
|
|
|
|
|
- name: DIETClassifier
|
|
|
|
|
epochs: 100
|
|
|
|
|
learning_rate: 0.001
|
|
|
|
|
tensorboard_log_directory: ./log
|
|
|
|
|
- name: ResponseSelector
|
|
|
|
|
epochs: 100
|
|
|
|
|
learning_rate: 0.001
|
|
|
|
|
- name: FallbackClassifier
|
|
|
|
|
threshold: 0.4
|
|
|
|
|
ambiguity_threshold: 0.1
|
|
|
|
|
- name: EntitySynonymMapper
|
|
|
|
|
|
|
|
|
|
policies:
|
|
|
|
|
- name: MemoizationPolicy
|
|
|
|
|
- name: TEDPolicy
|
|
|
|
|
- name: RulePolicy
|
|
|
|
|
core_fallback_threshold: 0.87
|
|
|
|
|
core_fallback_action_name: "action_default_fallback"
|
|
|
|
|
enable_fallback_prediction: True
|
|
|
|
|
- name: MemoizationPolicy
|
|
|
|
|
- name: TEDPolicy
|
|
|
|
|
- name: RulePolicy
|
|
|
|
|
core_fallback_threshold: 0.4
|
|
|
|
|
core_fallback_action_name: "action_default_fallback"
|
|
|
|
|
enable_fallback_prediction: True
|
|
|
|
|