增大test_size

ai/EstimateMarketTrendV20.py

@@ -22,7 +22,7 @@ uploadData = False  # 开启会上传预测结果
 doReport = True  # 开启会生成Excel报告
 
 # 待预测指数
-PREDICT_LIST = [156]
+PREDICT_LIST = [67]
 # PREDICT_LIST = [67, 121, 122, 123, 147, 148, 149, 150, 151, 152, 153, 154, 155, 156, 157, 158, 159, 160, 161, 162, 163,
 #                 164, 165, 166, 167, 168, 169, 170, 171, 174, 175, 177, 178]
 eco = [65, 66, 74, 134, 191]

ai/training_data_builder.py

@@ -185,7 +185,7 @@ class TrainingDataBuilder(ABC):
             date_index = []
         else:
             # Step 2: Split data into train set and test set
-            X_train, X_test, y_train, y_test = train_test_split(scaledX, y, test_size=0.02, shuffle=False)
+            X_train, X_test, y_train, y_test = train_test_split(scaledX, y, test_size=0.04, shuffle=False)
             date_index = DataAll['date'][-len(X_test):-self._numForecastDays].to_numpy()
             # To avoid data leak, test set should start from numForecastDays later
             X_test = X_test[:-self._numForecastDays]