|
@@ -119,9 +119,9 @@ def similarity_rank(movie_videos, sim_videos):
|
|
|
dist = cos_dist(vec1, vec2)
|
|
dist = cos_dist(vec1, vec2)
|
|
|
if dist > 0:
|
|
if dist > 0:
|
|
|
# item_sim[vid] = dist
|
|
# item_sim[vid] = dist
|
|
|
- item_sim = {'top_video_id': video_id, 'vid': vid, 'dist': dist}
|
|
|
|
|
|
|
+ item_sim = {'top_video_id': video_id, 'title': title, 'vid': vid, 'title1': title1 'dist': dist}
|
|
|
sim_result.append(item_sim)
|
|
sim_result.append(item_sim)
|
|
|
- dist_df = pd.DataFrame(sim_result, columns=['top_video_id', 'vid', 'dist'])
|
|
|
|
|
|
|
+ dist_df = pd.DataFrame(sim_result, columns=['top_video_id', 'title', 'vid', 'title1', 'dist'])
|
|
|
dist_df.to_csv('./data/videos_dist.csv', index=False)
|
|
dist_df.to_csv('./data/videos_dist.csv', index=False)
|
|
|
# sim_result[video_id] = item_sim
|
|
# sim_result[video_id] = item_sim
|
|
|
# print(video_id, item_sim)
|
|
# print(video_id, item_sim)
|
