bigbio/meddocan
Viewer • Updated • 2k • 473 • 3
How to use IIC/XLM_R_Galen-meddocan with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("token-classification", model="IIC/XLM_R_Galen-meddocan") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("IIC/XLM_R_Galen-meddocan")
model = AutoModelForSequenceClassification.from_pretrained("IIC/XLM_R_Galen-meddocan")This model is a finetuned version of XLM_R_Galen for the meddocan dataset used in a benchmark in the paper TODO. The model has a F1 of 0.947
Please refer to the original publication for more information TODO LINK
| parameter | Value |
|---|---|
| batch size | 16 |
| learning rate | 4e-05 |
| classifier dropout | 0.2 |
| warmup ratio | 0 |
| warmup steps | 0 |
| weight decay | 0 |
| optimizer | AdamW |
| epochs | 10 |
| early stopping patience | 3 |
TODO