Browse Source

WordCrfDTO 设置性别和年龄

kwzbigdata 4 years ago
parent
commit
dddbac0526

+ 8 - 0
src/main/java/com/diagbot/dto/WordCrfDTO.java

@@ -12,6 +12,14 @@ import lombok.Data;
  */
 @Data
 public class WordCrfDTO {
+    /**
+     * 年龄
+     */
+    private Integer age;
+    /**
+     * 性别(1:男,2:女,3:通用)
+     */
+    private Integer sex;
     // 主诉
     private ChiefLabel chiefLabel;
     // 现病史

+ 3 - 0
src/main/java/com/diagbot/facade/CommonFacade.java

@@ -29,8 +29,11 @@ public class CommonFacade {
 
     //组装好的label
     public WordCrfDTO crf_process(SearchData searchData){
+
         AIAnalyze aiAnalyze = new AIAnalyze(crfServiceClient);
         WordCrfDTO wordCrfDTO = new WordCrfDTO();
+        wordCrfDTO.setAge(searchData.getAge());
+        wordCrfDTO.setSex(searchData.getSex());
         aiAnalyze.aiProcess(searchData, wordCrfDTO);
         return wordCrfDTO;
     }