|
|
@@ -1,5 +1,6 @@
|
|
|
import datetime
|
|
|
import pandas as pd
|
|
|
+import os
|
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
@@ -13,4 +14,7 @@ if __name__ == '__main__':
|
|
|
data_df_list.append(cur_dt_df)
|
|
|
all_df = pd.concat(data_df_list)
|
|
|
print(f"all data num: {all_df.shape[0]}")
|
|
|
- all_df.to_csv(f'./data/train_data/train_{dt}.csv', index=False)
|
|
|
+ train_test_data_dir = './data/train_test_data'
|
|
|
+ if not os.path.exists(train_test_data_dir):
|
|
|
+ os.makedirs(train_test_data_dir)
|
|
|
+ all_df.to_csv(f'{train_test_data_dir}/train_test_{dt}.csv', index=False)
|
