search

gmichalo / UmlsBERT

MIT License
94 stars 16 forks source link

Pretrained model for MedNLI #10

Open griff4692 opened 2 years ago

griff4692 commented 2 years ago

Hi -

are there weights for a pre-trained MedNLI model?

gmichalo commented 2 years ago

Hello,

We do not provide the weights for each individual task but we have shared a notebook that can be used to train UmlsBERT on the MedNLI task: https://github.com/gmichalo/UmlsBERT/blob/master/experiments/MedNLI_task.ipynb