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

test: Cont'd debugging of keyword selection

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -156,11 +156,12 @@ def generate_plot(
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
  cluster_to_words_similarities = [(word_ix, similarity) for word_ix, similarity in enumerate(cluster_to_words_similarities)]
160
  cluster_to_words_similarities = sorted(cluster_to_words_similarities, key=lambda x: x[1], reverse=True)
 
161
  while len(cluster_keyword[label]) < 3:
162
  most_descriptive = cluster_to_words_similarities.pop(0)
163
- st.write(most_descriptive)
164
  cluster_keyword[label].append(cluster_words[most_descriptive[0]])
165
  cluster_keyword[label] = ', '.join(cluster_keyword[label])
166
  encoded_labels_keywords = [cluster_keyword[encoded_label] for encoded_label in encoded_labels]
 
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]])
166
  cluster_keyword[label] = ', '.join(cluster_keyword[label])
167
  encoded_labels_keywords = [cluster_keyword[encoded_label] for encoded_label in encoded_labels]