获取 1 月数据

functions/generate_data.py

@@ -137,19 +137,15 @@ class VideoDataGenerator(object):
 
 
 if __name__ == '__main__':
-    # print(generate_label_date("2024031101"))
-    date_list = generate_daily_strings("20240312", "20240319")
-    print(date_list)
+    date_list = generate_hourly_strings("2024010100", "2024013134")
     V = VideoDataGenerator()
-    # w = V.get_daily_data("20240319")
-    # print(w)
     L = {}
-    for date_str in tqdm(date_list):
-        L[date_str] = {}
-        data_list = V.get_daily_data(date_str)
-        for obj in tqdm(data_list):
-            video_id = obj['video_id']
-            L[date_str][video_id] = obj
-
-    with open('../data/12-17-train_label.json', 'w') as f:
+    # for date_str in tqdm(date_list):
+    #     L[date_str] = {}
+    #     data_list = V.get_hour_data(date_str)
+    #     for obj in tqdm(data_list):
+    #         video_id = obj['video_id']
+    #         L[date_str][video_id] = obj
+
+    with open('../data/january_train.json', 'w') as f:
         f.write(json.dumps(L, ensure_ascii=False, indent=4))