Forráskód Böngészése

generate label for mysql

罗俊辉 1 éve
szülő
commit
9fa30fd7fa
1 módosított fájl, 1 hozzáadás és 1 törlés
  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()