behnamsa commited on
Commit
1ec5925
1 Parent(s): fb7e0ab

Increase logits sensitivity, fix typo

Browse files
Files changed (1) hide show
  1. 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(-10*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
- [{'lable': word, 'score': score} for word, score in zip(top_words, top_scores)]
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
  ]