|
|
@@ -121,7 +121,7 @@ def similarity_rank(movie_videos, sim_videos):
|
|
|
# item_sim[vid] = dist
|
|
|
item_sim = {'top_video_id': video_id, 'vid': vid, 'dist': dist}
|
|
|
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', 'vid', 'dist'], index=False)
|
|
|
dist_df.to_csv('./data/videos_dist.csv')
|
|
|
# sim_result[video_id] = item_sim
|
|
|
# print(video_id, item_sim)
|
