train_config.json

{
  "model_name": "sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
  "max_length": 128,
  "epochs": 25,
  "batch_size": 16,
  "lr_encoder": 8e-06,
  "lr_heads": 0.0004,
  "weight_decay": 0.05,
  "task_weight_macro": 1.2,
  "task_weight_intent": 1.5,
  "task_weight_context": 0.8,
  "device": "cuda",
  "output_dir": "models\\artifacts",
  "seed": 42,
  "patience": 5,
  "min_delta": 0.001
}