소스 검색

模型入参增加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);
     }
 
     /**