linfan 1 år sedan
förälder
incheckning
1c1342c1d1
1 ändrade filer med 7 tillägg och 2 borttagningar
  1. 7 2
      video_rank.py

+ 7 - 2
video_rank.py

@@ -173,8 +173,13 @@ def video_new_rank(videoIds, fast_flow_set, flow_set, size, top_K, flow_pool_P):
     video_items = []
     for i in range(len(video_scores)):
         try:
-            video_score_str = json.load(video_scores[i])
-            video_items.append((videoIds[i], video_score_str[0]))
+            print(video_scores[i])
+            if video_scores[i] is None:
+                video_items.append((videoIds[i], 0.0))
+            else:
+                video_score_str = json.load(video_scores[i])
+                print("video_score_str:",video_score_str)
+                video_items.append((videoIds[i], video_score_str[0]))
         except Exception:
             video_items.append((videoIds[i], 0.0))
     sort_items = sorted(video_items, key=lambda k: k[1], reverse=True)