STRIDE-lab/pubmedbert-condition-20250221

pubmedbert_condition_20250221

Fine-tuned model from the PsyNamic project.

Base Model

microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract

Task

Condition

Training Parameters

{
  "mode": "train",
  "data": "<private_dataset>",
  "task": "Condition",
  "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
}