Modelpublic

STRIDE-lab/pubmedbert-study-control-20250221

sourceHugging Faceapache-2.0updated 3mo agoView on Hugging Face

0likes7downloads

Model Card

pubmedbertstudycontrol_20250221

Fine-tuned model from the PsyNamic project.

Base Model

microsoft/BiomedNLP-PubMedBERT-base-uncased-abstract

Task

Study Control

Training Parameters

json

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