bert-emotion / README.md
autoevaluator's picture
Add evaluation results on the emotion config and train split of tweet_eval
857d94a
|
raw
history blame
5.88 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
config: emotion
split: train
args: emotion
metrics:
- type: precision
value: 0.7071669427034283
name: Precision
- type: recall
value: 0.723286061789479
name: Recall
- type: accuracy
value: 0.9278477126189745
name: Accuracy
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiY2NmYWJlYTczYWIwZjBlZjYwZWI0Y2EwOTg1OWVjY2EzNzJhMzBjOGQ3NTlmYWIzY2VmY2Y3ZjE4MzljZmIzMCIsInZlcnNpb24iOjF9.xS7u6-CAg7KqAQZvAzyTk0FQxtXylw7J8kVuBJtlW7MzidFQ5SXWzJ-Vi_Kmn2rNVn4rIe_FWPGOHAaObIBTCQ
- type: f1
value: 0.9158678000901822
name: F1 Macro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiZDQzMDdlODZmZjVmM2MxMmU0ZmU3MDdkYTE0MjdkMjVhNjk3NmIzMmNkNWNlMjg2NDgyOGVmZTBjZTZkNjJkMiIsInZlcnNpb24iOjF9.5Ct17t887aDP7r97jkHQZJ8XcjrodAqs7OtxvS5A8PxeTR0b1CIrpuLHdR120-rFGjjgHngb7rGC_MgfTBBXAQ
- type: f1
value: 0.9278477126189745
name: F1 Micro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiOTg1MzkwOTE5MGIzNWUzYzViNTA4YTg1NzJkODZhMDdjZjcyMTczZDVmMWFkNzg5YzZlNmM4NTQwMmVjYjk0YiIsInZlcnNpb24iOjF9.LuJl7FquqtTolpt9Omfzf9FD2etgX4U25sHQMeaHUgOa122_wkYvxhvX4Tv3ZkD_qzsSH7Dl8wV4KYr91ZpcCA
- type: f1
value: 0.9273444254664317
name: F1 Weighted
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNjQ0ZTVkYjVhZTJkYjJkODgwNjIwYTBkMDIyZWZkNWRjNzAzYTk2ZTI1MmY5NDUzZWJhYzA3YTc4NzhkZDlhYiIsInZlcnNpb24iOjF9.owMV_PsPZGYBtvyaPx9xbAmAIoYdGu6FGhdD1GLfJQ9MgYIqZyixZmxZACLCmLY0nliiLAvK3M43k_fu4JBoDg
- type: precision
value: 0.9235213794497246
name: Precision Macro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiZDQ0Zjg3NDExYzFkZDU1MWI3Y2VmYjYyYzA1NGQ4NTZlMzE0MGU4MDJiYjFhZWJhNDA0MTFlOWZhOWNjYjdhMSIsInZlcnNpb24iOjF9.m6TD-vn17m_qtDDyWyVdL8DFP8uO8F2tdsBHIidAKDhCcI3J6rRyBGkxG_zDTXCCaCaCmdL4AJlUsA9JVOx9Aw
- type: precision
value: 0.9278477126189745
name: Precision Micro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiNDczMDgzZWQ5ZDZhMzZmM2U4ZjY4OWJmNjNjMTVjN2U0N2RjNzZlNTk2MWE5ZDIxNGIwODg5MmFhZjc3Y2IwYyIsInZlcnNpb24iOjF9.Itjh_jg1p_Ribol_x2nyetiL1aKq6yiBF5jgX7sAlytHTBRrzA6PVmQ8KY0GCyak6kxyx7Br_5Fp-VP5YgXGAw
- type: precision
value: 0.9297044059730123
name: Precision Weighted
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiYWMxMGE4M2RiOThiYzkxNmVmNDYzYTg4N2I1MzRiMjMxMWU1OWRlMjc2YjA3MjczMDNmMDRjMTJlMjIwYjVlOSIsInZlcnNpb24iOjF9._gLrUwqouatSr6Sxb9vTzRnmY_L_43JRdbfmf0p8P6_NdxqpTFsgBpc1YlUj_LjriwMGcmeZGdeB23ALIGl_BA
- type: recall
value: 0.9105403010148454
name: Recall Macro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiZjA1ZjNjYjJjNjBkMDAwOTFhMGEwMDgzMThjMWVkZWExNWZlMzg0ZWQ5MzRlMjU1ZWY2NzRlZGJkZDAzODk2MCIsInZlcnNpb24iOjF9.xAqeBqGJJEidya2s9-qMtyG0BZXpcNTm1WDFqAZ6GUr2sA0NfWTZIcciKBr2BuyShXfNFUcBugFyjJY3jIP2AA
- type: recall
value: 0.9278477126189745
name: Recall Micro
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiYjhiYTI1OTliOGZiYjc0MTdiYjZiMmVlNGYyYTRjZWYwMTI5YmEzNmE2M2JjNDhlNGQ5ZWZjNWM1MTk3M2ZlNyIsInZlcnNpb24iOjF9.LHh81wJR1AZnSVic5_rc1Bu7QgKtsh77VjVe95UbBAMJRr3oiH_dpcJAvHoXqwCZxV3pXR5AXEMzOPCv8BiTCw
- type: recall
value: 0.9278477126189745
name: Recall Weighted
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMGQ0NGI5MmJiNWFlY2U5ZWQ5YWIzYzFjY2EyZGI1Zjk1Y2Y4OTBjOWE0MjFhYmI3MzlkNjUwNDNhOWJiZmVmMiIsInZlcnNpb24iOjF9.OYJ5yWBDFQNBTXBg0yuvru8OMipcRBE_jn9OiXIO-hyMDp-HBF34FfBEPdN20pojDbPTnhjo36Zr7SbREx9uBA
- type: loss
value: 0.25382938981056213
name: loss
verified: true
verifyToken: eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJoYXNoIjoiMTNkZmM0NjgzNzMwY2IyMjg4NTE1NWZkN2M0ZmZjMTFhOGEwOTFiZTRiZWFjZTc2ODA2MDM4MzFmMTMwZGFkOCIsInZlcnNpb24iOjF9.LHVB3oBpgPJ0DGCidrNqnFFj-PMRt1V-_Z2undLWlCVEMVNfl0OGbczLBHxmTQvrRe_2ID8tyrwGBC8Os7kaBw
---
<!-- 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.2053
- Precision: 0.7072
- Recall: 0.7233
- Fscore: 0.7124
## 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.8623 | 1.0 | 815 | 0.7198 | 0.7536 | 0.6312 | 0.6559 |
| 0.5637 | 2.0 | 1630 | 0.8756 | 0.7213 | 0.7166 | 0.7160 |
| 0.2845 | 3.0 | 2445 | 1.2053 | 0.7072 | 0.7233 | 0.7124 |
### Framework versions
- Transformers 4.25.1
- Pytorch 1.13.0+cu116
- Datasets 2.8.0
- Tokenizers 0.13.2