获取 1 月数据

functions/generate_data.py

@@ -140,13 +140,13 @@ if __name__ == '__main__':
     date_list = generate_hourly_strings("2024010100", "2024013123")
     V = VideoDataGenerator()
     L = {}
-    print(date_list)
-    # 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
+    # print(date_list)
+    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))