浏览代码

generate label for mysql

罗俊辉 1 年之前
父节点
当前提交
9fa30fd7fa
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      main_spider.py

+ 1 - 1
main_spider.py

@@ -173,7 +173,7 @@ class LightGBM(object):
         print("统计 score 信息")
         print(data_series.describe())
         # 评估模型
-        accuracy = accuracy_score(Y_test, y_pred_binary)
+        accuracy = accuracy_score(y, y_pred_binary)
         print(f"Accuracy: {accuracy}")
         fw.close()