You need to agree to share your contact information to access this model

This repository is publicly accessible, but you have to accept the conditions to access its files and content.

Log in or Sign Up to review the conditions and access this model content.

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
Downloads last month
0
Inference Examples
This model does not have enough activity to be deployed to Inference API (serverless) yet. Increase its social visibility and check back later, or deploy to Inference Endpoints (dedicated) instead.

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