bert-ai-paper-classifier-arxiv

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

Classifications

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.

Usage

from transformers import pipeline

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

text = "Paste your abstract here"

output = classifier(text)
output