Michael-Geis commited on
Commit
c8cfc43
1 Parent(s): 6cc55f5

fixed error in generate_tag_embeddings

Browse files
Files changed (1) hide show
  1. embedding.py +1 -1
embedding.py CHANGED
@@ -105,7 +105,7 @@ def generate_tag_embeddings(model_name, path_to_tag_dict, path_to_save_embedding
105
  dict_string = file.read()
106
  tag_dict = json.loads(dict_string)
107
 
108
- tag_name_list = list(tag_dict.values())
109
  embedded_tag_names = model.encode(sentences=tag_name_list, show_progress_bar=True)
110
  embedded_tag_names_df = pd.DataFrame(embedded_tag_names)
111
  embedded_tag_names_df.columns = [
 
105
  dict_string = file.read()
106
  tag_dict = json.loads(dict_string)
107
 
108
+ tag_name_list = list(set(tag_dict.values()))
109
  embedded_tag_names = model.encode(sentences=tag_name_list, show_progress_bar=True)
110
  embedded_tag_names_df = pd.DataFrame(embedded_tag_names)
111
  embedded_tag_names_df.columns = [