Merge pull request #81 from ebotiab/fix-crf-model

Fix CRF model to not to translate tags in predict
microsoft · Oct 11, 2023 · d105ee2 · d105ee2
2 parents 6745717 + 0175263
commit d105ee2
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]