|
@@ -174,7 +174,8 @@ class LightGBM(object):
|
|
|
real_label = y[index]
|
|
|
score = item
|
|
|
prid_label = y_pred_binary[index]
|
|
|
- print(real_label, "\t", prid_label, "\t", score)
|
|
|
+ if score > 0.6:
|
|
|
+ print(real_label, "\t", prid_label, "\t", score)
|
|
|
fw.write("{}\t{}\t{}\n".format(real_label, prid_label, score))
|
|
|
score_list.append(score)
|
|
|
print("预测样本总量: {}".format(len(score_list)))
|