分词

src/main/java/com/diagbot/facade/TestwordInfoFacade.java

@@ -64,17 +64,13 @@ public class TestwordInfoFacade extends TestwordInfoServiceImpl {
         AIAnalyze aiAnalyze = new AIAnalyze(crfServiceClient);
 
         // 症状标准词
-        List<String> symptomStand = new ArrayList<>();
-        // List<String> symptomStand = getStandWord(9, 10000).getSymptomNames();
+        List<String> symptomStand = getStandWord(9, 100000).getSymptomNames();
         // 诊断标准词
-        // List<String> diseaseStand = getStandWord(4, 10000).getDiseaseNames().stream().map(r -> r.getName()).collect(Collectors.toList());
-        List<String> diseaseStand = new ArrayList<>();
+        List<String> diseaseStand = getStandWord(4, 100000).getDiseaseNames().stream().map(r -> r.getName()).collect(Collectors.toList());
         // 药品标准词
-        List<String> drugStand = new ArrayList<>();
-        // List<String> drugStand = getStandWord(5, 10000).getDrugNames().stream().map(r -> r.getName()).collect(Collectors.toList());
+        List<String> drugStand = getStandWord(5, 100000).getDrugNames().stream().map(r -> r.getName()).collect(Collectors.toList());
         // 手术标准词
-        List<String> operateStand = new ArrayList<>();
-        // List<String> operateStand = getStandWord(6, 10000).getOperationNames().stream().map(r -> r.getName()).collect(Collectors.toList());
+        List<String> operateStand = getStandWord(6, 100000).getOperationNames().stream().map(r -> r.getName()).collect(Collectors.toList());
 
         // 模型处理数据
         WordCrfDTO wordCrfDTO = new WordCrfDTO();
@@ -109,10 +105,10 @@ public class TestwordInfoFacade extends TestwordInfoServiceImpl {
                         CoreUtil.addSet(drugList, CoreUtil.getName(wordCrfDTO.getPresentLabel().getMedicines()));
                         // 手术
                         CoreUtil.addSet(operateList, CoreUtil.getName(wordCrfDTO.getPresentLabel().getOperations()));
-                        // // 化验
-                        // CoreUtil.addSet(drugList, CoreUtil.getName(wordCrfDTO.getPresentLabel().getDiags()));
-                        // // 辅检
-                        // CoreUtil.addSet(diseaseList, CoreUtil.getName(wordCrfDTO.getPresentLabel().getDiags()));
+                        // 化验
+                        // CoreUtil.addSet(drugList, CoreUtil.getName(wordCrfDTO.getLis().getDiags()));
+                        // 辅检
+                        CoreUtil.addSet(diseaseList, CoreUtil.getName(wordCrfDTO.getPacsLabel().getRes()));
                         break;
                     case "既往史":
                         searchData.setPasts(text);
@@ -131,6 +127,10 @@ public class TestwordInfoFacade extends TestwordInfoServiceImpl {
                         CoreUtil.addSet(diseaseList, CoreUtil.getName(wordCrfDTO.getVitalLabel().getVitals()));
                         break;
                     case "辅助检查":
+                        searchData.setSymptom(text);
+                        aiAnalyze.aiProcess(searchData, wordCrfDTO);
+                        // 辅检
+                        CoreUtil.addSet(diseaseList, CoreUtil.getName(wordCrfDTO.getPacsLabel().getRes()));
                         break;
                     case "初步诊断":
                     case "修正诊断":
@@ -180,7 +180,7 @@ public class TestwordInfoFacade extends TestwordInfoServiceImpl {
         retrievalVO.setType(type);
         retrievalVO.setInputStr("");
         retrievalVO.setSize(size);
-        return retrievalFacade.index(retrievalVO);
+        return retrievalFacade.getname(retrievalVO);
     }
 
     /**