librarian-bots/arxiv-new-datasets-v3
Viewer • Updated • 7.54k • 94
How to use davanstrien/modernbert-arxiv-new-dataset-v3 with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="davanstrien/modernbert-arxiv-new-dataset-v3") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("davanstrien/modernbert-arxiv-new-dataset-v3")
model = AutoModelForSequenceClassification.from_pretrained("davanstrien/modernbert-arxiv-new-dataset-v3", device_map="auto")answerdotai/ModernBERT-base fine-tuned for single-label text classification on librarian-bots/arxiv-new-datasets-v3.
new_dataset, no_new_dataset| Metric | Value |
|---|---|
| accuracy | 0.9550 |
| f1_macro | 0.9511 |
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model = AutoModelForSequenceClassification.from_pretrained("davanstrien/modernbert-arxiv-new-dataset-v3", trust_remote_code=True)
tokenizer = AutoTokenizer.from_pretrained("davanstrien/modernbert-arxiv-new-dataset-v3", trust_remote_code=True)
inputs = tokenizer("your text here", return_tensors="pt", truncation=True)
print(model.config.id2label[model(**inputs).logits.argmax().item()])
Produced on Hugging Face Jobs (gpu) with the train-classifier.py recipe from uv-scripts. Run it yourself:
hf jobs uv run --flavor gpu --secrets HF_TOKEN \
https://huggingface.co/datasets/uv-scripts/classification/raw/main/train-classifier.py \
librarian-bots/arxiv-new-datasets-v3 davanstrien/modernbert-arxiv-new-dataset-v3 --model answerdotai/ModernBERT-base --label-column labels
Base model
answerdotai/ModernBERT-base