Spaces:
Runtime error
Runtime error
ajitrajasekharan
commited on
Commit
•
318ce43
1
Parent(s):
60014a7
Update BatchInference.py
Browse files- BatchInference.py +3 -3
BatchInference.py
CHANGED
@@ -530,9 +530,9 @@ class BatchInference:
|
|
530 |
#if (index in string.punctuation or len(index) == 1 or index.startswith('.') or index.startswith('[') or index.startswith("#")):
|
531 |
if (index in string.punctuation or len(index) == 1 or index.startswith('.') or index.startswith('[')):
|
532 |
continue
|
533 |
-
|
534 |
-
|
535 |
-
|
536 |
#print(index,round(float(sorted_d[index]),4))
|
537 |
if (sent_index % 2 != 0):
|
538 |
#CS predictions
|
|
|
530 |
#if (index in string.punctuation or len(index) == 1 or index.startswith('.') or index.startswith('[') or index.startswith("#")):
|
531 |
if (index in string.punctuation or len(index) == 1 or index.startswith('.') or index.startswith('[')):
|
532 |
continue
|
533 |
+
if (index.startswith("#")): #subwords suggest model is trying to predict a multi word term that generally tends to be noisy. So penalize. Count and skip
|
534 |
+
k += 1
|
535 |
+
continue
|
536 |
#print(index,round(float(sorted_d[index]),4))
|
537 |
if (sent_index % 2 != 0):
|
538 |
#CS predictions
|