Skip to content

Instantly share code, notes, and snippets.

@erap129

erap129/recsys_knn_item.py

Last active May 5, 2022 16:28
Show Gist options
  • Save erap129/6e92d4c1c81ff21acfb14236840f25f4 to your computer and use it in GitHub Desktop.
Save erap129/6e92d4c1c81ff21acfb14236840f25f4 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
recsys_knn_item.py
algo_kwargs = dict(k=50, sim_options={'name': 'pearson', 'user_based': False, 'verbose' : True})
get_algorithm_report(KNNBasicWithTqdm, trainset, testset, train_loocv, test_loocv, movies_df,
target_movie_id='movie_1', target_user_id='user_1', top_k=10,
algo_kwargs_trainset=algo_kwargs, algo_kwargs_trainset_loocv=algo_kwargs, calc_most_similar=True)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment