Shredder commited on
Commit
3e7f51b
·
1 Parent(s): 57a4ee5

Update score_fincat.py

Browse files
Files changed (1) hide show
  1. score_fincat.py +1 -0
score_fincat.py CHANGED
@@ -23,6 +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
  continue
27
  prediction = lr_clf.predict(features.reshape(1, 768))
28
  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))