Browse Source

疾病模型推送结果最多修改为6个,其他保持由外部传参控制

louhr 5 years ago
parent
commit
7312472e64

+ 3 - 0
bigdata-web/src/main/java/org/diagbot/bigdata/work/AlgorithmCore.java

@@ -53,6 +53,9 @@ public class AlgorithmCore {
             switch (searchData.getAlgorithmType() == null ? 1 : searchData.getAlgorithmType()) {
                 case 1: //机器学习算法推理
                     executor = AlgorithmFactory.getInstance(classifies[i]);
+                    if (FeatureType.parse(featureTypes[i]) == FeatureType.DIAG) {
+                        searchData.setLength(6);//模型推送最多6个比较合理
+                    }
                     break;
                 case 2: //朴素贝叶斯算法推理
                     if (FeatureType.parse(featureTypes[i]) == FeatureType.DIAG) {

+ 1 - 1
push-web/src/main/java/org/diagbot/push/controller/AlgorithmController.java

@@ -150,7 +150,7 @@ public class AlgorithmController extends BaseController {
         responseData.setCrisisDetails(crisisApplication.crisisContent(searchData));
 
         //大数据推送
-        searchData.setLength(6);    //模型推送最多6个比较合理
+//        searchData.setLength(6);    //模型推送最多6个比较合理
         AlgorithmCore core = new AlgorithmCore();
         ResponseData bigDataResponseData = core.algorithm(request, searchData, responseData);