Shredder commited on
Commit
b9e9269
1 Parent(s): 3bd74e1

Update score_fincat.py

Browse files
Files changed (1) hide show
  1. score_fincat.py +1 -1
score_fincat.py CHANGED
@@ -23,7 +23,7 @@ def score_fincat(txt):
23
  context_text = extract_context_words(x)
24
  features = bert_embedding_extract(context_text, word)
25
  if(features[0]=='None'):
26
- highlight.append((word, '')
27
  continue
28
  prediction = lr_clf.predict(features.reshape(1, 768))
29
  prediction_probability = '{:.4f}'.format(round(lr_clf.predict_proba(features.reshape(1, 768))[:,1][0], 4))
 
23
  context_text = extract_context_words(x)
24
  features = bert_embedding_extract(context_text, word)
25
  if(features[0]=='None'):
26
+ highlight.append((word, ''))
27
  continue
28
  prediction = lr_clf.predict(features.reshape(1, 768))
29
  prediction_probability = '{:.4f}'.format(round(lr_clf.predict_proba(features.reshape(1, 768))[:,1][0], 4))