wilmerags commited on
Commit
9499fb6
·
1 Parent(s): 26caed7

fix: Removing numpy coercion for keyword selection

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -155,11 +155,8 @@ def generate_plot(
155
  cluster_embeddings_avg = np.mean(cluster_embeddings, axis=0)
156
  cluster_words_embeddings = embed_text(cluster_words, model)
157
  cluster_to_words_similarities = util.dot_score(cluster_embeddings_avg, cluster_words_embeddings)
158
- cluster_to_words_similarities = cluster_to_words_similarities.numpy()
159
- st.write(cluster_to_words_similarities)
160
  cluster_to_words_similarities = [(word_ix, similarity) for word_ix, similarity in enumerate(cluster_to_words_similarities)]
161
  cluster_to_words_similarities = sorted(cluster_to_words_similarities, key=lambda x: x[1], reverse=True)
162
- st.write(cluster_to_words_similarities)
163
  while len(cluster_keyword[label]) < 3:
164
  most_descriptive = cluster_to_words_similarities.pop(0)
165
  cluster_keyword[label].append(cluster_words[most_descriptive[0]])
 
155
  cluster_embeddings_avg = np.mean(cluster_embeddings, axis=0)
156
  cluster_words_embeddings = embed_text(cluster_words, model)
157
  cluster_to_words_similarities = util.dot_score(cluster_embeddings_avg, cluster_words_embeddings)
 
 
158
  cluster_to_words_similarities = [(word_ix, similarity) for word_ix, similarity in enumerate(cluster_to_words_similarities)]
159
  cluster_to_words_similarities = sorted(cluster_to_words_similarities, key=lambda x: x[1], reverse=True)
 
160
  while len(cluster_keyword[label]) < 3:
161
  most_descriptive = cluster_to_words_similarities.pop(0)
162
  cluster_keyword[label].append(cluster_words[most_descriptive[0]])