Edit model card


This model is a fine-tuned version of bert-base-cased on the ArXiv dataset.


This fine-tuned language model, based on bert-base-cased, is designed to classify AI-related abstracts into distinct categories. The model has been trained to accurately assign abstracts to one of five categories:

  • "Machine Learning Algorithms"
  • "Natural Language Processing"
  • "Computer Vision and Image Processing"
  • "Robotics and Autonomous Systems"
  • or "Other"

By leveraging its understanding of language and contextual cues, this model offers a powerful tool for automated classification of AI abstracts, facilitating efficient organization and analysis of diverse research topics within the field.


from transformers import pipeline

classifier = pipeline("text-classification", "Kushtrim/bert-ai-paper-classifier-arxiv")

text = "Paste your abstract here"

output = classifier(text)
Downloads last month

Collection including Kushtrim/bert-ai-paper-classifier-arxiv