Hugging Face's logo

Spaces:
wilmerags
/
tweet-snest
Build error

App Files Community
1
wilmerags commited on
Commit
a4593ce
·
1 Parent(s): 9499fb6

test: debug similarity comparison

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -155,6 +155,7 @@ 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 = [(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:
 
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
+ st.write(cluster_to_words_similarities)
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: