Browse Source

add ad_generate_train_test.py

liqian 1 year ago
parent
commit
8e558b0240
2 changed files with 16 additions and 1 deletions
  1. 0 1
      ad_feature_process.py
  2. 16 0
      ad_generate_train_test.py

+ 0 - 1
ad_feature_process.py

@@ -26,7 +26,6 @@ features = [
     'mid_view_count_pv_30day',
     'mid_play_count_30day',
     'mid_play_count_pv_30day',
-
     'mid_share_count_30day',
     'mid_share_count_pv_30day',
     'mid_return_count_30day',

+ 16 - 0
ad_generate_train_test.py

@@ -0,0 +1,16 @@
+import datetime
+import pandas as pd
+
+
+if __name__ == '__main__':
+    now_date = datetime.datetime.today()
+    dt = datetime.datetime.strftime(now_date, '%Y%m%d')
+    data_df_list = []
+    for days in range(2, 9):
+        cur_dt = datetime.datetime.strftime(now_date - datetime.timedelta(days=days), '%Y%m%d')
+        print(f"cur_dt = {cur_dt}")
+        cur_dt_df = pd.read_csv(f"./data/sample_train_data/{cur_dt}.csv")
+        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)