pytorch distilbert text-classification en dataset:arxiv_dataset transformers license:apache-2.0
---
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.
