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.

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: -

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

BERT Models

Collection

By: Kushtrim Visoka. This is a collection of BERT models that I have fine-tuned for various tasks and in multiple languages. • 7 items • Updated Feb 25