From 67c2499106d25d606243d633402fd0def4885e60 Mon Sep 17 00:00:00 2001 From: Ben Cohen Date: Sun, 28 Jul 2019 15:33:19 -0400 Subject: 25 to 10 --- utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'utils.py') diff --git a/utils.py b/utils.py index 465b0dc..5e19a4f 100644 --- a/utils.py +++ b/utils.py @@ -89,7 +89,7 @@ def get_closest(beer_id): one_embed = final_data[beer_id]['embed'] beer_ids = [] - for thing in sorted(small_embeddings, key = lambda x: cosine(one_embed, x[1]), reverse=False)[:25]: + for thing in sorted(small_embeddings, key = lambda x: cosine(one_embed, x[1]), reverse=False)[:10]: if thing[0] in beer_names: beer_ids.append(thing[0]) -- cgit v1.2.3