Pārlūkot izejas kodu

增加疾病提取

louhr 6 gadi atpakaļ
vecāks
revīzija
66850ad92f

+ 6 - 1
nlp-web/src/main/java/org/diagbot/nlp/controller/FeatureController.java

@@ -64,7 +64,12 @@ public class FeatureController extends BaseController<Feature, FeatureWrapper, L
         Response response = new Response();
         //获取入参中的特征信息
         FeatureAnalyze fa = new FeatureAnalyze();
-        List<Map<String, Object>> symptomsList = fa.start(text, FeatureType.SYMPTOM);
+        List<Map<String, Object>> symptomsList_symptom = fa.start(text, FeatureType.SYMPTOM);
+        List<Map<String, Object>> symptomsList_diag = fa.start(text, FeatureType.DIAG);
+
+        List<Map<String, Object>> symptomsList = new ArrayList<>();
+        symptomsList.addAll(symptomsList_symptom);
+        symptomsList.addAll(symptomsList_diag);
         response.setData(symptomsList);
         return response;
     }