metadata

language:
  - en
datasets:
  - Arxiv
model-index:
  - name: bert-ai-paper-classifier-arxiv
    results: []
widget:
  - Paste your abstract here
pipeline_tag: text-classification

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