浏览代码

辅检优化,当辅检项目不存在时,结果不提取,结构化辅检除外

kongwz 4 年之前
父节点
当前提交
f868fa1da1

+ 3 - 3
src/main/java/com/diagbot/model/ai/BeHospitalizedAI.java

@@ -318,7 +318,7 @@ public class BeHospitalizedAI extends ModelAI {
             pacsLabel = new PacsLabel();
             chiefLabel.setPacsLabel(pacsLabel);
         }
-        entityProcessPacs.extractEntity(aiOut, pacsLabel);
+        entityProcessPacs.extractEntity(aiOut, pacsLabel,false);
         wordCrfDTO.setChiefLabel(chiefLabel);
     }
 
@@ -346,7 +346,7 @@ public class BeHospitalizedAI extends ModelAI {
             pacsLabel = new PacsLabel();
             presentLabel.setPacsLabel(pacsLabel);
         }
-        entityProcessPacs.extractEntity(aiOut, pacsLabel);
+        entityProcessPacs.extractEntity(aiOut, pacsLabel,false);
         //服用药品
         List<Medicine> medicines = takeMedicine(presentLabel);
         presentLabel.setTakeMedicine(medicines);
@@ -709,7 +709,7 @@ public class BeHospitalizedAI extends ModelAI {
             return;
         }
         EntityProcessPacs entityProcessPacs = new EntityProcessPacs();
-        entityProcessPacs.extractEntity(aiOut, pacsLabel);
+        entityProcessPacs.extractEntity(aiOut, pacsLabel,true);
         //        EntityProcessPacsNew entityProcessPacsNew = new EntityProcessPacsNew();
         // Map<String, PacsNew> map = entityProcessPacsNew.extractEntity(aiOut);
         //

+ 7 - 3
src/main/java/com/diagbot/model/ai/process/EntityProcessPacs.java

@@ -17,13 +17,17 @@ import java.util.stream.Collectors;
 
 public class EntityProcessPacs extends EntityProcess {
 
-    public void extractEntity(JSONObject aiOut, PacsLabel pacsLabel) {
+    public void extractEntity(JSONObject aiOut, PacsLabel pacsLabel, boolean flagPacs) {
+
         // 辅检项目
         List<Lemma> pacsAll = createEntityTree(aiOut, EntityEnum.AUXILIARY_EXAMINATION.toString());
         // 如果辅检项目为空,就不存在辅检提取的其他内容
-        if (ListUtil.isEmpty(pacsAll)) {
-            return ;
+        if(!flagPacs){
+            if (ListUtil.isEmpty(pacsAll)) {
+                return ;
+            }
         }
+
         List<Item> collect = Lists.newArrayList();
         if(ListUtil.isNotEmpty(pacsAll)) {
             collect = pacsAll.stream().map(x -> {