瀏覽代碼

模型入参增加sentence

louhr 5 年之前
父節點
當前提交
4c4b89748a
共有 1 個文件被更改,包括 5 次插入0 次删除
  1. 5 0
      common-push/src/main/java/org/diagbot/common/push/work/ParamsDataProxy.java

+ 5 - 0
common-push/src/main/java/org/diagbot/common/push/work/ParamsDataProxy.java

@@ -138,6 +138,11 @@ public class ParamsDataProxy {
             featuresList = fa.start(searchData.getDiag(), FeatureType.DIAG);
             paramFeatureInit(searchData, featuresList);
         }
+
+        //主诉、现病史、体征文本信息一并传入模型
+        Map<String, String> featureMap = new HashMap<>();
+        featureMap.put("sentence", searchData.getSymptom());
+        searchData.getInputs().put("sentence", featureMap);
     }
 
     /**