Increase logits sensitivity, fix typo
Browse files- pipeline.py +2 -2
pipeline.py
CHANGED
@@ -41,10 +41,10 @@ class PreTrainedPipeline():
|
|
41 |
similarities = distance.cdist(embeddings.reshape((1,300)), self.comparisons, "cosine")[0]
|
42 |
top_indices = similarities.argsort()[:10]
|
43 |
top_words = [self.id2h[str(top_indices[i])] for i in range(10)]
|
44 |
-
logits = np.exp(-
|
45 |
softmax_probs = tf.nn.softmax(logits).numpy()
|
46 |
top_scores = [round(float(softmax_probs[i]), 3) for i in range(10)]
|
47 |
|
48 |
return [
|
49 |
-
[{'
|
50 |
]
|
|
|
41 |
similarities = distance.cdist(embeddings.reshape((1,300)), self.comparisons, "cosine")[0]
|
42 |
top_indices = similarities.argsort()[:10]
|
43 |
top_words = [self.id2h[str(top_indices[i])] for i in range(10)]
|
44 |
+
logits = np.exp(-1000*np.array(similarities[top_indices]))
|
45 |
softmax_probs = tf.nn.softmax(logits).numpy()
|
46 |
top_scores = [round(float(softmax_probs[i]), 3) for i in range(10)]
|
47 |
|
48 |
return [
|
49 |
+
[{'label': word, 'score': score} for word, score in zip(top_words, top_scores)]
|
50 |
]
|