From 818706d7426c2a792b009f8d06d5d7d361712046 Mon Sep 17 00:00:00 2001 From: Ben Cohen Date: Sun, 28 Jul 2019 14:19:51 -0400 Subject: -> small --- utils.py | 31 +++---------------------------- 1 file changed, 3 insertions(+), 28 deletions(-) (limited to 'utils.py') diff --git a/utils.py b/utils.py index 589fa63..a9a0786 100644 --- a/utils.py +++ b/utils.py @@ -59,7 +59,6 @@ style_name_to_num = {y[0]: x for x,y in styles.items()} - attr_centers = {} for attr, rel_styles in attr_to_styles.items(): centers = [style_centers[x] for x in rel_styles] @@ -172,30 +171,6 @@ def translate_to_style(embedding, style): new_point = embedding + (vector_between/TRANSLATION_CONSTANT) return get_closest_to_point(new_point, style_limit=style) - -# In[17]: - - -# beer_id = '388-1703' # cantillio gueueze -# beer_id = '140-276' # SN pale ale -beer_id = '4-59' # allagash white -# beer_id = '64-33832' # palo salto marron - -info = beer_names[beer_id] -print('Real ABV: {}'.format(info[2])) - -one_embed = final_data[beer_id]['embed'] -print(translate_to_style(one_embed, 'Belgian Quadrupel (Quad)')) - - -# beer_id = '388-1703' # cantillio gueueze -# beer_id = '140-276' # SN pale ale -beer_id = '4-59' # allagash white -# beer_id = '64-33832' # palo salto marron - -info = beer_names[beer_id] -print('Real ABV: {}'.format(info[2])) - -one_embed = final_data[beer_id]['embed'] -print(translate_to_attr(one_embed, 'Hoppy', 1)) - +def get_drinks_like(beer_id): + data = final_data[beer_id] + return data['alc'] \ No newline at end of file -- cgit v1.2.3