Browse Source

添加手术、诊断相似度

kongwz 4 years ago
parent
commit
34e88aa414
1 changed files with 18 additions and 0 deletions
  1. 18 0
      src/main/java/com/diagbot/facade/NeoFacade.java

+ 18 - 0
src/main/java/com/diagbot/facade/NeoFacade.java

@@ -38,6 +38,12 @@ public class NeoFacade {
     DiseaseRepository diseaseRepository;
     DiseaseRepository diseaseRepository;
     @Autowired
     @Autowired
     MedicineCodeRepository medicineCodeRepository;
     MedicineCodeRepository medicineCodeRepository;
+    /**
+     * modelName
+     *chief_present 症状
+     *operation 手术
+     *diagnose 诊断
+     */
     @Autowired
     @Autowired
     ChiefPresentSimilarityServiceClient chiefPresentSimilarityServiceClient;
     ChiefPresentSimilarityServiceClient chiefPresentSimilarityServiceClient;
     @Autowired
     @Autowired
@@ -277,6 +283,18 @@ public class NeoFacade {
                 similarMap.get(StandConvertEnum.clinical.getName()), "chief_present");
                 similarMap.get(StandConvertEnum.clinical.getName()), "chief_present");
         map.put(StandConvertEnum.clinical.getName(), clinicalMap);
         map.put(StandConvertEnum.clinical.getName(), clinicalMap);
 
 
+        // 诊断数据
+        List<String> diag_list = standConvert.getDiaglList();
+        Map<String, String> diseaseMap = getStandConvertTypeMap(modelAI, diag_list,
+                similarMap.get(StandConvertEnum.disease.getName()), "diagnose");
+        map.put(StandConvertEnum.disease.getName(), diseaseMap);
+
+        // 手术数据
+        List<String> operation_list = standConvert.getOperationList();
+        Map<String, String> operationMap = getStandConvertTypeMap(modelAI, operation_list,
+                similarMap.get(StandConvertEnum.operation.getName()), "operation");
+        map.put(StandConvertEnum.operation.getName(), diseaseMap);
+
         return map;
         return map;
     }
     }