You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
30 lines
680 B
YAML
30 lines
680 B
YAML
|
|
recipe: default.v1
|
|
language: zh
|
|
|
|
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
|
|
|
|
policies:
|
|
- name: MemoizationPolicy
|
|
- name: TEDPolicy
|
|
- name: RulePolicy
|
|
core_fallback_threshold: 0.87
|
|
core_fallback_action_name: "action_default_fallback"
|
|
enable_fallback_prediction: True
|