|
@@ -22,4 +22,5 @@ if __name__ == '__main__':
|
|
|
# 测试集
|
|
|
test_dt = datetime.datetime.strftime(now_date - datetime.timedelta(days=2), '%Y%m%d')
|
|
|
test_df = pd.read_csv(f"./data/sample_train_data/{test_dt}.csv")
|
|
|
- all_df.to_csv(f'{train_test_data_dir}/test_{dt}.csv', index=False)
|
|
|
+ print(f"test data num: {test_df.shape[0]}")
|
|
|
+ test_df.to_csv(f'{train_test_data_dir}/test_{dt}.csv', index=False)
|