updated labels

This commit is contained in:
j-hartmann 2021-06-15 10:16:23 +00:00
parent f759ba9691
commit 8298d8a51e
1 changed files with 7 additions and 7 deletions

View File

@ -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,