Browse Source

第一个科室概率大于0.3,第二个科室概率小于0.15,只推第一个

zhaops 5 years ago
parent
commit
bb03d5d76a
1 changed files with 12 additions and 2 deletions
  1. 12 2
      triage-service/src/main/java/com/diagbot/facade/AIFacade.java

+ 12 - 2
triage-service/src/main/java/com/diagbot/facade/AIFacade.java

@@ -342,14 +342,24 @@ public class AIFacade {
                             continue;
                             continue;
                         }
                         }
                         //第二个科室概率小于15%,剔除
                         //第二个科室概率小于15%,剔除
-                        if (items.size() == 1 && Double.valueOf(syfDiseaseDeptDTO.getRate()) < 0.15d) {
+                        /*if (items.size() == 1 && Double.valueOf(syfDiseaseDeptDTO.getRate()) < 0.15d) {
                             break;
                             break;
-                        }
+                        }*/
                         items.add(syfDiseaseDeptDTO);
                         items.add(syfDiseaseDeptDTO);
                     }
                     }
                 }
                 }
+                if (items.size() == 2) {
+                    //第一个科室概率小于0.3,两个科室都推
+                    //第一个科室概率大于0.3,第二个科室概率大于等于0.15,两个都推
+                    //第一个科室概率大于0.3,第二个科室概率小于0.15,只推第一个
+                    if (Double.valueOf(items.get(0).getRate()) > 0.3d
+                            && Double.valueOf(items.get(1).getRate()) < 0.15d) {
+                        items.remove(1);
+                    }
+                }
             }
             }
         }
         }
+        //按大科室分组展示
         Map<String, List<SYFDiseaseDeptDTO>> itemMap = EntityUtil.makeEntityListMap(items, "deptName");
         Map<String, List<SYFDiseaseDeptDTO>> itemMap = EntityUtil.makeEntityListMap(items, "deptName");
         List<SYFDeptDTO> syfDeptDTOList = Lists.newLinkedList();
         List<SYFDeptDTO> syfDeptDTOList = Lists.newLinkedList();
         itemMap.entrySet().forEach(item -> {
         itemMap.entrySet().forEach(item -> {