liqian 1 gadu atpakaļ
vecāks
revīzija
051bf2c147
1 mainītis faili ar 2 papildinājumiem un 1 dzēšanām
  1. 2 1
      ad_generate_train_test.py

+ 2 - 1
ad_generate_train_test.py

@@ -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)