Update score_fincat.py
Browse files- 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))
|