소스 검색

仅通过标题tag 分析全部数据

罗俊辉 1 년 전
부모
커밋
e3c9319bb1
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      main.py

+ 1 - 1
main.py

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