diff options
Diffstat (limited to 'utils.py')
-rw-r--r-- | utils.py | 31 |
1 files changed, 3 insertions, 28 deletions
@@ -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 |