gonzalocordova commited on
Commit
9b41375
·
1 Parent(s): 2a72c85

fix: somefax was applied twice

Browse files
Files changed (2) hide show
  1. NLPutils.py +0 -2
  2. app.py +0 -1
NLPutils.py CHANGED
@@ -242,7 +242,5 @@ def sentence_cats_probs(words, topic_centroids, model, topic_to_idx):
242
  for cat in topic_centroids.keys():
243
  d = distance.euclidean(cent, topic_centroids[cat])
244
  result[topic_to_idx[cat]] = -d
245
- # softmax
246
- result = np.exp(result) / np.sum(np.exp(result), axis=0)
247
 
248
  return result
 
242
  for cat in topic_centroids.keys():
243
  d = distance.euclidean(cent, topic_centroids[cat])
244
  result[topic_to_idx[cat]] = -d
 
 
245
 
246
  return result
app.py CHANGED
@@ -44,7 +44,6 @@ def predict_fn(text):
44
  for i in range(len(probs)):
45
  result[id_to_topic[i]] = probs[i]
46
 
47
- # Set to 0 if not in top 3 topics (sorted by probability)
48
  # sort result by value (highest to lowest)
49
  result = dict(sorted(result.items(), key=lambda item: item[1], reverse=True))
50
  # get top 3 topics dict
 
44
  for i in range(len(probs)):
45
  result[id_to_topic[i]] = probs[i]
46
 
 
47
  # sort result by value (highest to lowest)
48
  result = dict(sorted(result.items(), key=lambda item: item[1], reverse=True))
49
  # get top 3 topics dict