|
@@ -131,11 +131,11 @@ def similarity_rank(movie_videos, sim_videos):
|
|
|
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])
|
|
|
+ sim_result.extend(item_sim_list[:config_.SIM_N_19])
|
|
|
# to_redis
|
|
|
key_name = f"{config_.MOVIE_RELEVANT_LIST_KEY_NAME_PREFIX}{video_id}"
|
|
|
relevant_data = dict()
|
|
|
- for item in item_sim_list:
|
|
|
+ for item in item_sim_list[:config_.SIM_N_19]:
|
|
|
relevant_data[item['vid']] = item['dist']
|
|
|
if redis_helper.key_exists(key_name=key_name):
|
|
|
redis_helper.del_keys(key_name=key_name)
|