wilmerags commited on
Commit
d447a02
·
1 Parent(s): 29060ca

test: Try different element deletion of element in keyword selection

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -156,7 +156,7 @@ def generate_plot(
156
  cluster_to_words_similarities = cluster_to_words_similarities.numpy()
157
  while len(cluster_keyword[label]) < 3:
158
  most_descriptive = np.argmax(cluster_to_words_similarities)
159
- del cluster_to_words_similarities[most_descriptive]
160
  cluster_keyword[label].append(cluster_words[most_descriptive])
161
  encoded_labels_keywords = [cluster_keyword[encoded_label] for encoded_label in encoded_labels]
162
  embeddings_2d = get_tsne_embeddings(embeddings)
 
156
  cluster_to_words_similarities = cluster_to_words_similarities.numpy()
157
  while len(cluster_keyword[label]) < 3:
158
  most_descriptive = np.argmax(cluster_to_words_similarities)
159
+ cluster_to_words_similarities = np.delete(cluster_to_words_similarities, most_descriptive)
160
  cluster_keyword[label].append(cluster_words[most_descriptive])
161
  encoded_labels_keywords = [cluster_keyword[encoded_label] for encoded_label in encoded_labels]
162
  embeddings_2d = get_tsne_embeddings(embeddings)