获取 1 , 2月 label

test.py

@@ -34,9 +34,9 @@ def generate_train_label(item, y_ori_data):
     label_dt = generate_label_date(dt)
     label_obj = y_ori_data.get(label_dt, {}).get(video_id)
     if label_obj:
-        label = int(label_obj['total_return']) if label_obj['total_return'] else None
+        label = int(label_obj['total_return']) if label_obj['total_return'] else 0
     else:
-        label = None
+        label = 0
     return label, item_features
 
 
@@ -54,10 +54,10 @@ if __name__ == '__main__':
     for video_obj in tqdm(x_data):
         print(video_obj)
         our_label, features = generate_train_label(video_obj, y_data)
-        if our_label:
+        # if our_label:
 
-            x_list.append(features)
-            y_list.append(our_label)
+        x_list.append(features)
+        y_list.append(our_label)
     # print(len(y_list))
     with open("whole_data/x_data.json", "w") as f1:
         f1.write(json.dumps(x_list, ensure_ascii=False))