rasa : 修改config.yml

virtual-patient-rasa/docs/rasa/config-local.yml

@@ -1,29 +0,0 @@
-
-recipe: default.v1
-language: zh
-
-pipeline:
-  - name: JiebaTokenizer
-  - name: LanguageModelFeaturizer
-    model_name: "bert"
-    model_weights: "/rasa/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.4
-    core_fallback_action_name: "action_default_fallback"
-    enable_fallback_prediction: True

virtual-patient-rasa/docs/rasa/config.yml

@@ -5,8 +5,8 @@ language: zh
 pipeline:
   - name: JiebaTokenizer
   - name: LanguageModelFeaturizer
-    model_name: bert
+    model_name: "bert"
-    model_weights: bert-base-chinese
+    model_weights: "/rasa/bert-base-chinese"
   - name: RegexFeaturizer
   - name: DIETClassifier
     epochs: 100
@@ -16,14 +16,14 @@ pipeline:
     epochs: 100
     learning_rate: 0.001
   - name: FallbackClassifier
-    threshold: 0.87
+    threshold: 0.4
-    ambiguity_threshold: 0.1    
+    ambiguity_threshold: 0.1
   - name: EntitySynonymMapper
-  
+
 policies:
   - name: MemoizationPolicy
   - name: TEDPolicy
   - name: RulePolicy
-    core_fallback_threshold: 0.87
+    core_fallback_threshold: 0.4
     core_fallback_action_name: "action_default_fallback"
-    enable_fallback_prediction: True
+    enable_fallback_prediction: True