瀏覽代碼

add ad_generate_train_test.py

liqian 2 年之前
父節點
當前提交
8e558b0240
共有 2 個文件被更改,包括 16 次插入1 次删除
  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)