behnamsa
/

persian-reverse-dict

Text Classification

reverse-dictionary

Model card Files Files and versions Community

behnamsa commited on Mar 10, 2023

Commit

63fceef

•

1 Parent(s): e6e161e

Fix scores

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -41,7 +41,7 @@ class PreTrainedPipeline():
         similarities = distance.cdist(embeddings.reshape((1,300)), self.comparisons, "cosine")[0]
         top_indices = similarities.argsort()[:10]
         top_words = [self.id2h[str(top_indices[i])] for i in range(10)]
-        logits = np.exp(-1000*np.array(similarities[top_indices]))
         softmax_probs = tf.nn.softmax(logits).numpy()
         top_scores = [round(float(softmax_probs[i]), 3) for i in range(10)]

         similarities = distance.cdist(embeddings.reshape((1,300)), self.comparisons, "cosine")[0]
         top_indices = similarities.argsort()[:10]
         top_words = [self.id2h[str(top_indices[i])] for i in range(10)]
+        logits = -8*np.array(similarities[top_indices])
         softmax_probs = tf.nn.softmax(logits).numpy()
         top_scores = [round(float(softmax_probs[i]), 3) for i in range(10)]