metadata
license: apache-2.0
datasets:
- amazon_polarity
base_model: distilbert-base-uncased
model-index:
- name: distilbert-base-uncased-finetuned-emotion-balanced
results:
- task:
type: text-classification
name: Text Classification
dataset:
name: amazon_polarity
type: sentiment
args: default
metrics:
- type: accuracy
value: 0.958
name: Accuracy
- type: loss
value: 0.119
name: Loss
- type: f1
value: 0.957
name: F1
metrics:
- accuracy
- f1
pipeline_tag: text-classification
This model is a fine-tuned version of distilbert-base-uncased on a subset of the amazon-polarity dataset.