Ajay1311 commited on
Commit
b229894
·
verified ·
1 Parent(s): cc1928c

Update phish_config.json

Browse files
Files changed (1) hide show
  1. phish_config.json +13 -7
phish_config.json CHANGED
@@ -1,21 +1,27 @@
1
  {
2
- "_name_or_path": "bert-base-uncased",
3
  "architectures": ["BertForSequenceClassification"],
4
  "attention_probs_dropout_prob": 0.1,
5
  "classifier_dropout": null,
6
  "gradient_checkpointing": false,
7
  "hidden_act": "gelu",
8
  "hidden_dropout_prob": 0.1,
9
- "hidden_size": 768, // Changed to 768 for bert-base-uncased
10
- "id2label": { "0": "benign", "1": "phishing" },
 
 
 
11
  "initializer_range": 0.02,
12
- "intermediate_size": 3072, // Changed to 3072 for bert-base-uncased
13
- "label2id": { "benign": 0, "phishing": 1 },
 
 
 
14
  "layer_norm_eps": 1e-12,
15
  "max_position_embeddings": 512,
16
  "model_type": "bert",
17
- "num_attention_heads": 12, // Changed to 12 for bert-base-uncased
18
- "num_hidden_layers": 12, // Changed to 12 for bert-base-uncased
19
  "pad_token_id": 0,
20
  "position_embedding_type": "absolute",
21
  "problem_type": "single_label_classification",
 
1
  {
2
+ "_name_or_path": "bert-large-uncased",
3
  "architectures": ["BertForSequenceClassification"],
4
  "attention_probs_dropout_prob": 0.1,
5
  "classifier_dropout": null,
6
  "gradient_checkpointing": false,
7
  "hidden_act": "gelu",
8
  "hidden_dropout_prob": 0.1,
9
+ "hidden_size": 1024,
10
+ "id2label": {
11
+ "0": "benign",
12
+ "1": "phishing"
13
+ },
14
  "initializer_range": 0.02,
15
+ "intermediate_size": 4096,
16
+ "label2id": {
17
+ "benign": 0,
18
+ "phishing": 1
19
+ },
20
  "layer_norm_eps": 1e-12,
21
  "max_position_embeddings": 512,
22
  "model_type": "bert",
23
+ "num_attention_heads": 16,
24
+ "num_hidden_layers": 24,
25
  "pad_token_id": 0,
26
  "position_embedding_type": "absolute",
27
  "problem_type": "single_label_classification",