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