|
@@ -166,7 +166,7 @@ class LightGBM(object):
|
|
|
y_pred = bst.predict(x, num_iteration=bst.best_iteration)
|
|
|
pred_score_df = pd.DataFrame(list(y_pred), columns=['pred_score'])
|
|
|
temp = sorted(list(y_pred))
|
|
|
- yuzhi = temp[int(len(temp) * 0.5) - 1]
|
|
|
+ yuzhi = temp[int(len(temp) * 0.8) - 1]
|
|
|
y_pred_binary = [0 if i <= yuzhi else 1 for i in list(y_pred)]
|
|
|
pred_label_df = pd.DataFrame(list(y_pred_binary), columns=['pred_label'])
|
|
|
score_list = []
|