updated labels
This commit is contained in:
parent
599d32d232
commit
c0cc7f81e0
29
config.json
29
config.json
|
@ -10,17 +10,26 @@
|
|||
"hidden_act": "gelu",
|
||||
"hidden_dropout_prob": 0.1,
|
||||
"hidden_size": 768,
|
||||
"id2label": [
|
||||
"anger",
|
||||
"anger",
|
||||
"anger",
|
||||
"anger",
|
||||
"anger",
|
||||
"anger",
|
||||
"anger"
|
||||
],
|
||||
"id2label": {
|
||||
"0": "anger",
|
||||
"1": "disgust",
|
||||
"2": "fear",
|
||||
"3": "joy",
|
||||
"4": "neutral",
|
||||
"5": "sadness",
|
||||
"6": "surprise"
|
||||
},
|
||||
"initializer_range": 0.02,
|
||||
"intermediate_size": 3072,
|
||||
"label2id": {
|
||||
"anger": 0,
|
||||
"disgust": 1,
|
||||
"fear": 2,
|
||||
"joy": 3,
|
||||
"neutral": 4,
|
||||
"sadness": 5,
|
||||
"surprise": 6
|
||||
},
|
||||
"layer_norm_eps": 1e-05,
|
||||
"max_position_embeddings": 514,
|
||||
"model_type": "roberta",
|
||||
|
@ -33,4 +42,4 @@
|
|||
"type_vocab_size": 1,
|
||||
"use_cache": true,
|
||||
"vocab_size": 50265
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue