updated labels
This commit is contained in:
parent
f759ba9691
commit
8298d8a51e
14
config.json
14
config.json
|
@ -22,13 +22,13 @@
|
||||||
"initializer_range": 0.02,
|
"initializer_range": 0.02,
|
||||||
"intermediate_size": 3072,
|
"intermediate_size": 3072,
|
||||||
"label2id": {
|
"label2id": {
|
||||||
"LABEL_0": 0,
|
"anger": 0,
|
||||||
"LABEL_1": 1,
|
"disgust": 1,
|
||||||
"LABEL_2": 2,
|
"fear": 2,
|
||||||
"LABEL_3": 3,
|
"joy": 3,
|
||||||
"LABEL_4": 4,
|
"neutral": 4,
|
||||||
"LABEL_5": 5,
|
"sadness": 5,
|
||||||
"LABEL_6": 6
|
"surprise": 6
|
||||||
},
|
},
|
||||||
"layer_norm_eps": 1e-05,
|
"layer_norm_eps": 1e-05,
|
||||||
"max_position_embeddings": 514,
|
"max_position_embeddings": 514,
|
||||||
|
|
Loading…
Reference in New Issue