Skip to content

Commit 83728cc

Browse files
committed
chore: switch default PII to LoRA with threshold 0.9
Update main config to use LoRA PII detector with optimized threshold, consistent with e2e test profiles. Signed-off-by: Yossi Ovadia <[email protected]>
1 parent fe43b07 commit 83728cc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

config/config.yaml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -63,9 +63,9 @@ classifier:
6363
use_cpu: true
6464
category_mapping_path: "models/lora_intent_classifier_bert-base-uncased_model/category_mapping.json"
6565
pii_model:
66-
model_id: "models/pii_classifier_modernbert-base_presidio_token_model"
67-
use_modernbert: true
68-
threshold: 0.7
66+
model_id: "models/lora_pii_detector_bert-base-uncased_model"
67+
use_modernbert: false
68+
threshold: 0.9
6969
use_cpu: true
7070
pii_mapping_path: "models/pii_classifier_modernbert-base_presidio_token_model/pii_type_mapping.json"
7171

0 commit comments

Comments
 (0)