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))
|