mrm8488
/

bert-mini-finetuned-age_news-classification

Text Classification

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

Safetensors

Model size

11.2M params

Tensor type

I64

·

F32

·

Inference Examples

Text Classification

Inference API (serverless) is not available, repository is disabled.

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

View on Papers With Code