Spaces:
Build error
Build error
fix: Removing numpy coercion for keyword selection
Browse files
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]])
|