Spaces:
Sleeping
Sleeping
Arko Banik
commited on
Commit
·
c47440e
1
Parent(s):
8243f22
returns tuple of cat_idx and score
Browse files
app.py
CHANGED
|
@@ -214,7 +214,7 @@ def get_sorted_cosine_similarity(embeddings_metadata):
|
|
| 214 |
cat = categories[index]
|
| 215 |
cat_embed = category_embeddings[cat]
|
| 216 |
# Calc cosine sim
|
| 217 |
-
cat_scores.append((
|
| 218 |
# Store doc_id and score as a tuple
|
| 219 |
|
| 220 |
|
|
|
|
| 214 |
cat = categories[index]
|
| 215 |
cat_embed = category_embeddings[cat]
|
| 216 |
# Calc cosine sim
|
| 217 |
+
cat_scores.append((index, np.dot(input_embedding,cat_embed)))
|
| 218 |
# Store doc_id and score as a tuple
|
| 219 |
|
| 220 |
|