ml4pubmed

university

AI & ML interests

We like to decide where sentences belong.

PubMed Text Classification: The Models


Sharing a selection of fine-tuned and trained models on article section text classification on the PubMed dataset. Part of assignment 2 for Machine Learning for Healthcare @ ETH Zürich.

  • An interactive EDA of the model performance we observed can be found here
  • GitHub repo containing all the code/notebooks here

Peter Szemraj & Lou Ancillon