소스 검색

user_train_data

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

+ 1 - 1
main_userupload.py

@@ -94,7 +94,7 @@ class LightGBM(object):
         return features, labels
 
     def best_params(self):
-        path = "data/train_data/spider_train_20240402"
+        path = "data/train_data/spider_train_20240402.json"
         X, y = self.read_data(path)
         X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)