cduesing
/

bert_workforce

Text Classification

text-embeddings-inference

Model card Files Files and versions

cduesing commited on Jul 15, 2022

Commit

f3e06ef

·

1 Parent(s): 2beda03

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -23,7 +23,7 @@
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",
-  "problem_type": "regression",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 0,

   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",
+  "problem_type": "single_label_classification",
   "num_attention_heads": 12,
   "num_hidden_layers": 12,
   "pad_token_id": 0,