hinglish-prompt-injection-detector/config.yaml at main · kyahikaru/hinglish-prompt-injection-detector · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
# ============================================
# Hinglish Prompt Injection Detector (Final)
# Production Configuration
# ============================================

data:
  train_path: "data/srinivasan_dataset.xlsx"
  test_250_path: "data/hinglish-stealth-250.csv"
  test_110_path: "data/hinglish-stealth-110-heldout.csv"
  test_clean_path: "data/hinglish-clean-500.csv"
  test_size: 0.20
  random_state: 42

embeddings:
  model_name: "paraphrase-multilingual-MiniLM-L12-v2"
  model_path: "paraphrase-multilingual-MiniLM-L12-v2"

classifier:
  model_path: "models/final_classifier.onnx"
  model_type: "svm-rbf"
  probability_threshold: 0.5

rules:
  enabled: true

contextual_guard:
  enabled: true
  override_strategy: "classifier_safe_only"

normalization:
  enabled: true
  max_repeated_characters: 2

decision:
  strategy: "hybrid_layered"
  confidence_threshold_high: 0.90
  confidence_threshold_low: 0.50