How to use from the
Use from the
Transformers library
# Use a pipeline as a high-level helper
from transformers import pipeline

pipe = pipeline("text-classification", model="alfonsusrr/BugBERTa-C")
# Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification

tokenizer = AutoTokenizer.from_pretrained("alfonsusrr/BugBERTa-C")
model = AutoModelForSequenceClassification.from_pretrained("alfonsusrr/BugBERTa-C")
Quick Links

This model is trained on base model microsoft/codebert-base added with classification head and trained with DebugBench dataset for bug classification purpose.

Downloads last month
4
Safetensors
Model size
0.1B params
Tensor type
F32
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Dataset used to train alfonsusrr/BugBERTa-C