Update acrobert.py
Browse files- acrobert.py +1 -1
acrobert.py
CHANGED
@@ -82,7 +82,7 @@ def acrobert(sentence, model, device):
|
|
82 |
results = dict()
|
83 |
for acronym in rulebased_pairs.keys():
|
84 |
if rulebased_pairs[acronym][0] != '':
|
85 |
-
results
|
86 |
else:
|
87 |
pred, scores = predict(5, model, acronym, sentence, batch_size=10, acronym_kb=kb, device=device)
|
88 |
output = list(zip(pred, scores))
|
|
|
82 |
results = dict()
|
83 |
for acronym in rulebased_pairs.keys():
|
84 |
if rulebased_pairs[acronym][0] != '':
|
85 |
+
results[acronym] = rulebased_pairs[acronym][0]
|
86 |
else:
|
87 |
pred, scores = predict(5, model, acronym, sentence, batch_size=10, acronym_kb=kb, device=device)
|
88 |
output = list(zip(pred, scores))
|