{
  "mode": "train",
  "data": "<private_dataset>",
  "task": "Regimen",
  "model": "pubmedbert",
  "cross_val": false,
  "batch_size": 8,
  "learning_rate": 5e-05,
  "weight_decay": 0.01,
  "lr_scheduler": "linear",
  "warmup_ratio": 0.1,
  "epochs": 30,
  "dropout": 0.1,
  "early_stopping_patience": 5,
  "gradient_clipping": 0.1,
  "device": "cuda",
  "load": null,
  "max_length": 512,
  "is_multilabel": true
}

STRIDE-lab/pubmedbert-regimen-20250221

pubmedbert_regimen_20250221

Base Model

Task

Training Parameters

STRIDE-lab/
pubmedbert-regimen-20250221