harshkh/cyber-attack-classifier

1{
2  "_name_or_path": "./attack_classifier_model",
3  "activation": "gelu",
4  "architectures": [
5    "DistilBertForSequenceClassification"
6  ],
7  "attention_dropout": 0.1,
8  "dim": 768,
9  "dropout": 0.1,
10  "hidden_dim": 3072,
11  "id2label": {
12    "0": "SQL_Injection",
13    "1": "Directory_Traversal",
14    "2": "XSS",
15    "3": "File_Inclusion",
16    "4": "XXE"
17  },
18  "initializer_range": 0.02,
19  "label2id": {
20    "Directory_Traversal": 1,
21    "File_Inclusion": 3,
22    "SQL_Injection": 0,
23    "XSS": 2,
24    "XXE": 4
25  },
26  "max_position_embeddings": 512,
27  "model_type": "distilbert",
28  "n_heads": 12,
29  "n_layers": 6,
30  "pad_token_id": 0,
31  "problem_type": "single_label_classification",
32  "qa_dropout": 0.1,
33  "seq_classif_dropout": 0.2,
34  "sinusoidal_pos_embds": false,
35  "tie_weights_": true,
36  "torch_dtype": "float32",
37  "transformers_version": "4.40.0",
38  "vocab_size": 30522
39}
40