---
title: README
emoji: 📉
colorFrom: green
colorTo: blue
sdk: static
pinned: false
---

# **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](https://pubmed-ml4hc-transformers.netlify.app/)**
- GitHub repo containing all the code/notebooks **[here](https://github.com/pszemraj/pubmed-text-classification)**
---

Peter Szemraj & Lou Ancillon