Sfoglia il codice sorgente

1-debug,调整序列数据的时候和注释print函数。

bijl 5 anni fa
parent
commit
2c4a722c1b

+ 1 - 3
algorithm/src/main/java/org/algorithm/core/neural/dataset/NNDataSet.java

@@ -31,7 +31,7 @@ public abstract class NNDataSet {
     protected final Map<String, Map<String, Integer>> RELATED_DIAGNOSIS_DICT = new HashMap<>();  // 特征与疾病相关表
     private boolean doFilterDiagnosis = false;  // 是否做疾病过滤
 
-    private final float firstRateThreshold = 0.15f;  // 第一个疾病的概率阈值
+    private final float firstRateThreshold = 0.1f;  // 第一个疾病的概率阈值
     private final float rateSumThreshold = 0.6f;  // 概率和阈值
     private final int numToPush = 3;  // 推荐推送的个数
     private final float rapidFallTimes = 5;  // 骤降倍数
@@ -309,9 +309,7 @@ public abstract class NNDataSet {
      */
     public Map<String, Float> wrap(float[][] predict) {
         if (this.doFilterDiagnosis)  // 过滤疾病
-//            r
         {
-            this.wrapAndFilter(predict);
             return this.wrapAndFilterWithPushCountFilter(predict);
         } else
             return this.basicWrap(predict);