Edit model card

DistilBERT ArXiv Category Classification

DistilBERT model fine-tuned on a small subset of the ArXiv dataset to predict the category of a given paper.

Downloads last month
177
Inference API
This model can be loaded on Inference API (serverless).

Dataset used to train Wi/arxiv-distilbert-base-cased