diff options
author | Ben Cohen <ben@kensho.com> | 2019-07-28 15:31:44 -0400 |
---|---|---|
committer | Ben Cohen <ben@kensho.com> | 2019-07-28 15:31:44 -0400 |
commit | 2a77fb8d847f27b8f68156acceb23df0cdbe5059 (patch) | |
tree | 0420f0a9d973ae603ae31458ad2163f7a7151509 | |
parent | 4641b4354a0f8211d0d22226d53b95d41699e0bb (diff) |
-> style
-rw-r--r-- | utils.py | 7 |
1 files changed, 4 insertions, 3 deletions
@@ -122,7 +122,8 @@ def get_closest_to_point(one_embed, style_limit=[]): -def translate_to_attr(embedding, to_attr, amt): +def translate_to_attr(beer_id, to_attr, amt): + embedding = final_data[beer_id]['embed'] if amt < 0: back = True amt = abs(amt) @@ -157,8 +158,8 @@ def translate_to_attr(embedding, to_attr, amt): return get_closest_to_point(new_point) -def translate_to_style(embedding, style): - +def translate_to_style(beer_id, style): + embedding = final_data[beer_id]['embed'] closest_center_vector = style_centers[style] vector_between = closest_center_vector - embedding |