FelipeCasali-USP
/

lgpd_pii_identifier

Token Classification

Inference Endpoints

Model card Files Files and versions Community

FelipeCasali-USP commited on Aug 30, 2023

Commit

61d244e

•

1 Parent(s): 421b5ca

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,7 +1,7 @@
 {
   "_name_or_path": "lgpd_pii_identifier",
   "architectures": [
-    "BertForSequenceClassification"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
@@ -34,7 +34,7 @@
   "pooler_num_attention_heads": 12,
   "pooler_num_fc_layers": 3,
   "pooler_size_per_head": 128,
-  "problem_type": "single_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.25.1",
   "type_vocab_size": 2,

 {
   "_name_or_path": "lgpd_pii_identifier",
   "architectures": [
+    "DistilBertModel"
   ],
   "attention_probs_dropout_prob": 0.1,
   "classifier_dropout": null,
   "pooler_num_attention_heads": 12,
   "pooler_num_fc_layers": 3,
   "pooler_size_per_head": 128,
+  "problem_type": "multi_label_classification",
   "torch_dtype": "float32",
   "transformers_version": "4.25.1",
   "type_vocab_size": 2,