louhr преди 6 години
родител
ревизия
66850ad92f
променени са 1 файла, в които са добавени 6 реда и са изтрити 1 реда
  1. 6 1
      nlp-web/src/main/java/org/diagbot/nlp/controller/FeatureController.java

+ 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;
     }