unitary commited on
Commit
f78ed32
1 Parent(s): e9c286a

fix wrong activation function for multi-label classification

Browse files
Files changed (1) hide show
  1. config.json +1 -0
config.json CHANGED
@@ -47,6 +47,7 @@
47
  "female": 8,
48
  "homosexual_gay_or_lesbian": 9
49
  },
 
50
  "layer_norm_eps": 1e-05,
51
  "max_position_embeddings": 514,
52
  "model_type": "roberta",
 
47
  "female": 8,
48
  "homosexual_gay_or_lesbian": 9
49
  },
50
+ "problem_type": "multi_label_classification",
51
  "layer_norm_eps": 1e-05,
52
  "max_position_embeddings": 514,
53
  "model_type": "roberta",