liqian 2 years ago
parent
commit
605b05a0e4
1 changed files with 3 additions and 2 deletions
  1. 3 2
      videos_similarity.py

+ 3 - 2
videos_similarity.py

@@ -87,8 +87,9 @@ def get_movie_video_top_list():
     for index, row in data_df.iterrows():
         if row['videoid'] == 0:
             continue
-        if index < 20:
-            movie_videos[int(row['videoid'])] = row['title']
+        # if index > 20:
+        #     break
+        movie_videos[int(row['videoid'])] = row['title']
     return movie_videos