Wi's picture
Update README.md
001bb3a
metadata
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 to predict the category of a given paper.