Clinical STS Regression Model

Fine-tuned from cross-encoder/nli-deberta-v3-large on a clinical text-pair similarity dataset.

Training config

{
  "train": "./med_sts/clinicalSTS.train.txt",
  "test": "./med_sts/clinicalSTS.test.txt",
  "model": [
    "cross-encoder/nli-deberta-v3-large"
  ],
  "epochs": 20,
  "batch_size": 16,
  "lr": 2e-05,
  "weight_decay": 0.01,
  "warmup_ratio": 0.1,
  "max_grad_norm": 1.0,
  "dropout": 0.1,
  "max_len": 256,
  "seed": 42,
  "patience": 5,
  "output_dir": "./sts_val/nli-deberta-v3-large",
  "wandb_project": "clinical-sts",
  "hf_repo": "youssefkhalil320/nli-deberta-v3-large"
}

Best test Spearman ρ: 0.7891

Downloads last month
13
Safetensors
Model size
0.4B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for youssefkhalil320/nli-deberta-v3-large

Finetuned
(4)
this model