README.md

metadata

language: en
tags:
  - news
  - classification
  - mini
datasets:
  - ag_news
widget:
  - text: >-
      Israel withdraws from Gaza camp Israel withdraws from Khan Younis refugee
      camp in the Gaza Strip, after a four-day operation that left 11 dead.
model-index:
  - name: mrm8488/bert-mini-finetuned-age_news-classification
    results:
      - task:
          type: text-classification
          name: Text Classification
        dataset:
          name: ag_news
          type: ag_news
          config: default
          split: test
        metrics:
          - name: Accuracy
            type: accuracy
            value: 0.9339473684210526
            verified: true
          - name: Precision Macro
            type: precision
            value: 0.9341278202108704
            verified: true
          - name: Precision Micro
            type: precision
            value: 0.9339473684210526
            verified: true
          - name: Precision Weighted
            type: precision
            value: 0.9341278202108704
            verified: true
          - name: Recall Macro
            type: recall
            value: 0.9339473684210526
            verified: true
          - name: Recall Micro
            type: recall
            value: 0.9339473684210526
            verified: true
          - name: Recall Weighted
            type: recall
            value: 0.9339473684210526
            verified: true
          - name: F1 Macro
            type: f1
            value: 0.9339351653217216
            verified: true
          - name: F1 Micro
            type: f1
            value: 0.9339473684210526
            verified: true
          - name: F1 Weighted
            type: f1
            value: 0.9339351653217215
            verified: true
          - name: loss
            type: loss
            value: 0.20814141631126404
            verified: true

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

Test set accuray: 0.93