| 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. | |