diff options
author | Ben Cohen <ben@kensho.com> | 2019-07-28 15:33:19 -0400 |
---|---|---|
committer | Ben Cohen <ben@kensho.com> | 2019-07-28 15:33:19 -0400 |
commit | 67c2499106d25d606243d633402fd0def4885e60 (patch) | |
tree | e07f0ccd16a23ccdca934b990e3d0581ad1f60ad /utils.py | |
parent | 2a77fb8d847f27b8f68156acceb23df0cdbe5059 (diff) |
25 to 10
Diffstat (limited to 'utils.py')
-rw-r--r-- | utils.py | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -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]) |