emotion-classification-model

README.md

@@ -33,13 +33,13 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 3e-05
+- learning_rate: 5e-05
 - train_batch_size: 256
 - eval_batch_size: 256
 - seed: 42
 - optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 6
+- num_epochs: 5
 
 ### Framework versions
 

config.json

@@ -30,7 +30,6 @@
   "n_heads": 12,
   "n_layers": 6,
   "pad_token_id": 0,
-  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ddcdc9f950175b4af6555487001109213050401915f608abcb701e11d998ddb2
+oid sha256:80ef010dba86fbaabf5e583f510fa7e6bbc7b4fca1db2aa224bd69145569f0e9
 size 267844872

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d96a223d806604a58cac97900a4ca5c5364ee9f82306a147b55d9d3865ce1025
+oid sha256:6df76a334168e1aaa383c000952d944ec5b73cbc3306dbcc70783215af3e5963
 size 5240