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