speech-test commited on
Commit
a2ec440
1 Parent(s): 1d385e3

Fix labels

Browse files
Files changed (1) hide show
  1. config.json +12 -1
config.json CHANGED
@@ -50,8 +50,20 @@
50
  "hidden_dropout": 0.1,
51
  "hidden_dropout_prob": 0.1,
52
  "hidden_size": 768,
 
 
 
 
 
 
53
  "initializer_range": 0.02,
54
  "intermediate_size": 3072,
 
 
 
 
 
 
55
  "layer_norm_eps": 1e-05,
56
  "layerdrop": 0.1,
57
  "mask_feature_length": 10,
@@ -64,7 +76,6 @@
64
  "num_conv_pos_embeddings": 128,
65
  "num_feat_extract_layers": 7,
66
  "num_hidden_layers": 12,
67
- "num_labels": 4,
68
  "pad_token_id": 0,
69
  "problem_type": "single_label_classification",
70
  "torch_dtype": "float32",
 
50
  "hidden_dropout": 0.1,
51
  "hidden_dropout_prob": 0.1,
52
  "hidden_size": 768,
53
+ "id2label": {
54
+ "0": "neu",
55
+ "1": "hap",
56
+ "2": "ang",
57
+ "3": "sad"
58
+ },
59
  "initializer_range": 0.02,
60
  "intermediate_size": 3072,
61
+ "label2id": {
62
+ "ang": 2,
63
+ "hap": 1,
64
+ "neu": 0,
65
+ "sad": 3
66
+ },
67
  "layer_norm_eps": 1e-05,
68
  "layerdrop": 0.1,
69
  "mask_feature_length": 10,
 
76
  "num_conv_pos_embeddings": 128,
77
  "num_feat_extract_layers": 7,
78
  "num_hidden_layers": 12,
 
79
  "pad_token_id": 0,
80
  "problem_type": "single_label_classification",
81
  "torch_dtype": "float32",