---
license: apache-2.0
language: 
  - en
datasets:
- arxiv_dataset
tags:
- distilbert
---

# DistilBERT ArXiv Category Classification

DistilBERT model fine-tuned on a small subset of the [ArXiv dataset](https://www.kaggle.com/datasets/Cornell-University/arxiv) to predict the category of a given paper.