mrm8488
/

bert-mini-finetuned-age_news-classification

Text Classification Transformers PyTorch JAX Safetensors

English bert news classification mini Eval Results Inference Endpoints

Model card Files Files and versions Community

Edit model card

BERT-Mini fine-tuned on age_news dataset for news classification

Test set accuray: 0.93

Downloads last month: 326

Safetensors

Model size

11.2M params

Tensor type

I64

·

F32

·

Dataset used to train mrm8488/bert-mini-finetuned-age_news-classification

Spaces using mrm8488/bert-mini-finetuned-age_news-classification 3

Evaluation results

Accuracy on ag_news
test set verified

0.934
Precision Macro on ag_news
test set verified

0.934
Precision Micro on ag_news
test set verified

0.934
Precision Weighted on ag_news
test set verified

0.934
Recall Macro on ag_news
test set verified

0.934
Recall Micro on ag_news
test set verified

0.934
Recall Weighted on ag_news
test set verified

0.934
F1 Macro on ag_news
test set verified

0.934
F1 Micro on ag_news
test set verified

0.934
F1 Weighted on ag_news
test set verified

0.934
loss on ag_news
test set verified

0.208

View on Papers With Code