Skip to content

Commit de3790a

Browse files
committed
chore: switch default config to use LoRA models with optimized thresholds
Update default configuration to use LoRA-based classification: - Intent classification: lora_intent_classifier_bert-base-uncased_model - PII detection: lora_pii_detector_bert-base-uncased_model with threshold 0.9 This aligns the default config with e2e test configurations for consistency across all environments. Signed-off-by: Yossi Ovadia <[email protected]>
1 parent 52ec739 commit de3790a

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

config/config.yaml

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -58,15 +58,14 @@ model_config:
5858
# Classifier configuration
5959
classifier:
6060
category_model:
61-
model_id: "models/category_classifier_modernbert-base_model"
62-
use_modernbert: true
61+
model_id: "models/lora_intent_classifier_bert-base-uncased_model"
6362
threshold: 0.6
6463
use_cpu: true
65-
category_mapping_path: "models/category_classifier_modernbert-base_model/category_mapping.json"
64+
category_mapping_path: "models/lora_intent_classifier_bert-base-uncased_model/category_mapping.json"
6665
pii_model:
67-
model_id: "models/pii_classifier_modernbert-base_presidio_token_model"
68-
use_modernbert: true
69-
threshold: 0.7
66+
model_id: "models/lora_pii_detector_bert-base-uncased_model"
67+
use_modernbert: false
68+
threshold: 0.9
7069
use_cpu: true
7170
pii_mapping_path: "models/pii_classifier_modernbert-base_presidio_token_model/pii_type_mapping.json"
7271

0 commit comments

Comments
 (0)