Преглед на файлове

优化数据处理代码

罗俊辉 преди 1 година
родител
ревизия
a0d5e567af
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      main_spider.py

+ 1 - 1
main_spider.py

@@ -118,7 +118,7 @@ class LightGBM(object):
         Load dataset
         :return:
         """
-        path = "data/train_data/spider_data_20240402.json"
+        path = "data/train_data/spider_train_20240402.json"
         x, y = self.read_data(path)
         train_size = int(len(x) * self.split_c)
         X_train, X_test = x[:train_size], x[train_size:]