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