优化模板阈值

dev_2.1.0
liu 1 year ago
parent e7f6d69454
commit 14588a639a

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

Loading…
Cancel
Save