--- 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 --- # 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