Fix CRF model to not to translate tags in predict

microsoft · Oct 9, 2023 · 0175263 · 0175263
1 parent 6745717
commit 0175263
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/presidio_evaluator/models/crf_model.py b/presidio_evaluator/models/crf_model.py
@@ -94,9 +94,7 @@ def predict(self, sample: InputSample) -> List[str]:
 
     @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]