|
|
@@ -126,8 +126,8 @@ def similarity_rank(movie_videos, sim_videos):
|
|
|
# item_sim[vid] = dist
|
|
|
item_sim = {'top_video_id': video_id, 'title': title, 'vid': vid, 'title1': title1, 'dist': dist}
|
|
|
item_sim_list.append(item_sim)
|
|
|
- item_sim_list.sort(key=lambda x: x['dist'], reverse=True)
|
|
|
- sim_result.extend(item_sim_list[:4])
|
|
|
+ item_sim_list.sort(key=lambda x: x['dist'], reverse=True)
|
|
|
+ sim_result.extend(item_sim_list[:4])
|
|
|
# sim_result.append(item_sim)
|
|
|
dist_df = pd.DataFrame(sim_result, columns=['top_video_id', 'title', 'vid', 'title1', 'dist'])
|
|
|
dist_df.to_csv('./data/videos_dist.csv', index=False)
|
