bert-emotion / README.md
autoevaluator's picture
Add evaluation results on the emotion config and test split of tweet_eval
fb89d41
|
raw
history blame
6.03 kB
---
license: apache-2.0
tags:
- generated_from_trainer
datasets:
- tweet_eval
metrics:
- precision
- recall
model-index:
- name: bert-emotion
results:
- task:
type: text-classification
name: Text Classification
dataset:
name: tweet_eval
type: tweet_eval
args: emotion
metrics:
- type: precision
value: 0.7262254187805659
name: Precision
- type: recall
value: 0.725549671319356
name: Recall
- task:
type: text-classification
name: Text Classification
dataset:
name: tweet_eval
type: tweet_eval
config: emotion
split: test
metrics:
- type: accuracy
value: 0.7825475017593244
name: Accuracy
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiZTZhOTE5NjRlODcxNzEzMGUzNTllMWFlNjRkNzRmNTRmYmU1YjU2MDE1YWIyZDcwYTdjZGI0MjIxMDQ2OTgwMCIsInZlcnNpb24iOjF9.3D-oFO-DjRIHTob3DtWsu_hsLNted2Av5eZLg8LoCYIi_9oAcymIikHcSS4Hr-cDmpF4C7DsmaH3HPYdN-MJCw
- type: f1
value: 0.7373078325463487
name: F1 Macro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNDEwMTAyOWQ4ZjA0MGVhODFhODE3OGViNWU2MDM5YmM0ODNmMTVmYWE0Mjc4ZTI0YzE4NzIwODU1YWI4YWY1YSIsInZlcnNpb24iOjF9.CuhYKHUD7nI0Fcrh9ogQWqkM4iwvcdHoG1Xoi1fdcn7aD5CyVevqPKO4hxIzxpBcxaSaVFlkAa_rJzpB0pKqDw
- type: f1
value: 0.7825475017593244
name: F1 Micro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMWM1ODA3ZDJhYTJhMGJkYmU1ODYyZDNhMTM4ZTY3YmIwNjlhOGZiNDdlMWU2ZTg5MzgwYWM5YjA1OTAyZDA0NSIsInZlcnNpb24iOjF9.092oASK0jrq4j-pgNwXOCmKpR0s57T2aAIJepLJnFUd7wie3bmoT7MqlrALqGifrYe-t8UlSHdmSGNUrWfIFBA
- type: f1
value: 0.7800585587569513
name: F1 Weighted
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNmU4MTIyMGI1YjNlNmJhNTQ1MDdjZjVjOGNiNWVkZjk4NmRiZDczNDIwOWYyZWI4MTExNmQwZWY3NGU3N2RjYiIsInZlcnNpb24iOjF9.LqDQwwJoDAt4wTQmGWv5sZfZjFLTr2dBG6LtIXwA7m4SsOER6GKg1L1RmlkHRrHgTxBrMgF9WbSQOCdOJXByCg
- type: precision
value: 0.757828822693908
name: Precision Macro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiOTM1Yzg2ODY0ZTU0ZWJjODg1YjI4OGZkMDc3ZDMwZWY5ZTBhMDY4NDFmMTgxMmJiNDJjMGIyNWQ4YzFmZWVhYyIsInZlcnNpb24iOjF9.xbsTcqN3-vDUtoXd7by8F96pAfAel6aenJ5Vz5UvIxQmlI88orIg5DkrT_MIl8LOcIZkF9TFhz5RJGytnZNGBw
- type: precision
value: 0.7825475017593244
name: Precision Micro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNjM1ZjVlYjk1NWM0NzgyZTAxMTU3OWY3ZjBlYTJlZjcxMzViYzVjM2IzZDM4M2ZlZTA0MmZkYzFhMTAxMWViYSIsInZlcnNpb24iOjF9._hUo-MVxie-qEfO0QUKdJVwMV0dOoxkyGptEQoW7gx1r05p5FAFPt3vmuU9JjJlxhDaw-0seKooh39HU-7xyBw
- type: precision
value: 0.7817085293090495
name: Precision Weighted
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMWM0MWE0MmRiNzczMDY5YzdlOTZiY2Q2OWU5NWQ4MTA1M2U2YWMxZTlmZmIxYzg3MTgzNDgxZWUxNmJjYjdhNyIsInZlcnNpb24iOjF9.W7DlBHfESzF3lYrf4Buurq4Fxx1_YOzAEspsdkZr93H1Ih-DRRQaDZbk8KCrQOJaI83XzHfL_6XaQ_SyH3qpAg
- type: recall
value: 0.7268372321658545
name: Recall Macro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNDFiY2QxOGUxZDlkNGQ1NTM5ODljZjExZWM3ZGI0Y2YyN2E4NmU0Y2M0NWZiOGJjZTBmNGM2MWVmNzNiZWM3NCIsInZlcnNpb24iOjF9.Hig3mlWTnWSAKHouNistpgzwB5smK2piaUtf6AfeJEhUS-tUQYEbxQsK6Np_E799bCe-yi7bU_M9NZW8qAAcDg
- type: recall
value: 0.7825475017593244
name: Recall Micro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMGUzMjc4YTczYmFhZTI4ZTcwOTc1NmYwYWExMzFiOGIzOTBhNmQ2ZGNmZDc3NDljNDY0N2UwOGZlMGJhNmRlOSIsInZlcnNpb24iOjF9.ljZ9Wsn-942EqvfOVpAO6ZnWTHUdp0CbOMEGgJNeanXkk1jVL4rc7MQ6SSupEF_9jW2apl7no68CII__UUeiAw
- type: recall
value: 0.7825475017593244
name: Recall Weighted
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiYTIxZmQwMWNlOTljYjBhMzBjZTRjODFjODAxNWMwOTAyYWRhNWVjZTkxZDJjOTUxNzg4YjMxMDQ1MDFkNWMwMiIsInZlcnNpb24iOjF9.Q8V3qy0CLGZuWOAp5WE4eOGJT12zeDO0CRZByGtx2KyU8ikcQ_TZ1WVlUQgFVnWoiqqmkfsIEtyQBl6axEE5BQ
- type: loss
value: 0.8246609568595886
name: loss
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiZGQzZTk4YzM3OTI0MGUxYmY1YTU5YjUzOWIzMGVlN2Q1NTdjZmNhZjAxZGVmNTg3MTAyYzg3ODI1MmU3ZWM5ZSIsInZlcnNpb24iOjF9.feU19daDdE7OKkcop-Vq42FV2bZnHYVgBzsq6RM-823fZjdlgNZCWHFaEec2PF_HxIAnFDqsH6fJ7a1VhmTFCA
---
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
should probably proofread and complete it, then remove this comment. -->
# bert-emotion
This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on the tweet_eval dataset.
It achieves the following results on the evaluation set:
- Loss: 1.1670
- Precision: 0.7262
- Recall: 0.7255
- Fscore: 0.7253
## Model description
More information needed
## Intended uses & limitations
More information needed
## Training and evaluation data
More information needed
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 5e-05
- train_batch_size: 4
- eval_batch_size: 4
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 3
### Training results
| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | Fscore |
|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|
| 0.8561 | 1.0 | 815 | 0.7844 | 0.7575 | 0.6081 | 0.6253 |
| 0.5337 | 2.0 | 1630 | 0.9080 | 0.7567 | 0.7236 | 0.7325 |
| 0.2573 | 3.0 | 2445 | 1.1670 | 0.7262 | 0.7255 | 0.7253 |
### Framework versions
- Transformers 4.19.2
- Pytorch 1.11.0+cu113
- Datasets 2.2.2
- Tokenizers 0.12.1