rttl-ai
/

foody-bert

Text Classification

Inference Endpoints

Model card Files Files and versions Community

rttl commited on Sep 12, 2022

Commit

43d4306

•

1 Parent(s): 4da3470

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "RKSBertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
@@ -28,7 +28,7 @@
   "output_hidden_states": true,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
-  "problem_type": "single_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.17.0",
   "type_vocab_size": 2,

 {
   "architectures": [
+    "BertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
   "output_hidden_states": true,
   "pad_token_id": 0,
   "position_embedding_type": "absolute",
+  "problem_type": "multi_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.17.0",
   "type_vocab_size": 2,