Fix CRF model to not to translate tags in predict

This commit is contained in:
enrique.botia 2023-10-09 16:35:47 +02:00
Родитель 67457170b0
Коммит 0175263d85
1 изменённых файлов: 1 добавлений и 3 удалений

Просмотреть файл

@ -94,9 +94,7 @@ class CRFModel(BaseModel):
@staticmethod
def crf_predict(sample, model):
sample.translate_input_sample_tags()
conll = sample.to_conll(translate_tags=True)
conll = sample.to_conll(translate_tags=False)
sentence = [(di["text"], di["pos"], di["label"]) for di in conll]
features = CRFModel.sent2features(sentence)
return model.predict([features])[0]