YchKhan commited on
Commit
0e439ee
1 Parent(s): 316e9b1

Update classification.py

Browse files
Files changed (1) hide show
  1. classification.py +1 -1
classification.py CHANGED
@@ -87,7 +87,7 @@ def save_data(df, filename):
87
  df['Topic'] = df['Topic'].apply(lambda x: ', '.join(x) if isinstance(x, list) else x)
88
  df['Score'] = df['Score'].apply(lambda x: ', '.join(map(str, x)) if isinstance(x, list) else x)
89
 
90
-
91
  new_filename = filename.replace(".", "_classified.")
92
  df.to_excel(new_filename, index=False)
93
  return new_filename
 
87
  df['Topic'] = df['Topic'].apply(lambda x: ', '.join(x) if isinstance(x, list) else x)
88
  df['Score'] = df['Score'].apply(lambda x: ', '.join(map(str, x)) if isinstance(x, list) else x)
89
 
90
+ df = df.drop(columns=['Embeddings'])
91
  new_filename = filename.replace(".", "_classified.")
92
  df.to_excel(new_filename, index=False)
93
  return new_filename