Skip to content

Pretrained Models

Latest
Compare
Choose a tag to compare
@JohnGiorgi JohnGiorgi released this 25 Feb 21:43
· 39 commits to main since this release
89711f3

This release exists just to give us a place to host pretrained models. Please see the main branch for the latest source code.

  • cdr.tar.gz: Trained on the BC5CDR corpus train + validation sets without entity hints. Achieves a micro F1-score on the test set of 40.2%.
  • cdr_hints.tar.gz: Trained on the BC5CDR corpus train + validation sets with entity hints. Achieves a micro F1-score on the test set of 67.2%.
  • gda.tar.gz: Trained on the GDA corpus without entity hints. Achieves a micro F1-score on the test set of 55.2%.
  • gda_hints.tar.gz: Trained on the GDA corpus with entity hints. Achieves a micro F1-score on the test set of 84.9%.
  • dgm.tar.gz: Trained on the DGM corpus without entity hints. Achieves a micro F1-score on the test set of 64.7%.
  • dgm_hints.tar.gz: Trained on the DGM corpus with entity hints. Achieves a micro F1-score on the test set of 84.4%.
  • docred.tar.gz: Trained on the DocRED corpus without entity hints. Achieves a micro F1-score on the test set of 38.2%.

Please see our paper for more details on each pretrained model.