opentargets/clinical_trial_stop_reasons

<!-- This model card has been generated automatically according to the information the Trainer had access to. You should probably proofread and complete it, then remove this comment. -->

Clinical trial stop reasons

This model is a fine-tuned version of bert-base-uncased on the task of classification of why a clinical trial has stopped early.

The dataset containing 3,747 manually curated reasons used for fine-tuning is available in the Hub.

More details on the model training are available in the GitHub project (link) and in the associated publication (DOI: 10.1038/s41588-024-01854-z).

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

• —learning_rate: 5e-05
• —trainbatchsize: 32
• —evalbatchsize: 32
• —seed: 42
• —optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
• —lrschedulertype: linear
• —num_epochs: 7

Training results

Framework versions

• —Transformers 4.26.0
• —Pytorch 1.12.1+cu102
• —Datasets 2.9.0
• —Tokenizers 0.13.2