aboutsummaryrefslogtreecommitdiff
path: root/utils.py
diff options
context:
space:
mode:
authorBen Cohen <ben@kensho.com>2019-07-28 16:35:36 -0400
committerBen Cohen <ben@kensho.com>2019-07-28 16:35:36 -0400
commit7eeafc6dacc6173aa1a233acf4f39f18481b3843 (patch)
treec590c0ed6b4281e64e1c0eb1d6e92621c18d7f2c /utils.py
parent4f48801ef1ee4d3f5d199cde113748ecae64eddd (diff)
foo
Diffstat (limited to 'utils.py')
-rw-r--r--utils.py2
1 files changed, 1 insertions, 1 deletions
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])