rpg1
/

tinyBERT_phishing_model

Text Classification

Inference Endpoints

Model card Files Files and versions

rpg1 commited on Mar 1

Commit

f3f9322

•

1 Parent(s): 9773af2

Update config.json

Files changed (1) hide show

config.json +4 -0

config.json CHANGED Viewed

@@ -16,6 +16,10 @@
   },
   "initializer_range": 0.02,
   "intermediate_size": 1200,
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",

   },
   "initializer_range": 0.02,
   "intermediate_size": 1200,
+  "label2id": {
+    "benign": 0,
+    "phishing": 1
+  },
   "layer_norm_eps": 1e-12,
   "max_position_embeddings": 512,
   "model_type": "bert",