From 7eeafc6dacc6173aa1a233acf4f39f18481b3843 Mon Sep 17 00:00:00 2001 From: Ben Cohen Date: Sun, 28 Jul 2019 16:35:36 -0400 Subject: foo --- utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'utils.py') diff --git a/utils.py b/utils.py index 12e2312..3382d5e 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)[:10]: + for thing in sorted(small_embeddings, key = lambda x: cosine(one_embed, x[1]), reverse=False)[1:11]: if thing[0] in beer_names: beer_ids.append(thing[0]) -- cgit v1.2.3