Update classification.py

classification.py

@@ -178,7 +178,7 @@ def match_categories(df, category_df, treshold=0.45):
             cos_scores = util.cos_sim(ebd_content, torch.stack(list(category_df['Embeddings']), dim=0))[0]
             high_score_indices = [i for i, score in enumerate(cos_scores) if score > treshold]
             for j in high_score_indices:
-                df.loc[index, category_df.loc[j, 'topic']] = 'float(cos_scores[j])'
+                df.loc[index, category_df.loc[j, 'topic']] = float(cos_scores[j])
     return df
 
 def save_data(df, filename):