|
@@ -156,7 +156,7 @@ class LightGBM(object):
|
|
|
fw = open("result/summary_{}.txt".format(dt), "a+", encoding="utf-8")
|
|
|
path = 'data/predict_data/predict_{}.json'.format(dt)
|
|
|
x, y = self.read_data(path, yc=6)
|
|
|
- true_label_df = DataFrame(y, columns=['ture_label'])
|
|
|
+ true_label_df = DataFrame(list(y), columns=['ture_label'])
|
|
|
bst = lgb.Booster(model_file=self.model)
|
|
|
y_pred = bst.predict(x, num_iteration=bst.best_iteration)
|
|
|
pred_label_df = DataFrame(list(y_pred), columns=['pred_label'])
|