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