SharpAI commited on
Commit
0c6255f
·
1 Parent(s): 70677dc
Files changed (2) hide show
  1. config.json +86 -88
  2. tf_model.h5 +2 -2
config.json CHANGED
@@ -1,5 +1,4 @@
1
  {
2
- "_name_or_path": "SharpAI/mal_tls",
3
  "architectures": [
4
  "BertForSequenceClassification"
5
  ],
@@ -9,105 +8,104 @@
9
  "hidden_dropout_prob": 0.1,
10
  "hidden_size": 256,
11
  "id2label": {
12
- "0": "gootkit",
13
- "1": "emotet",
14
- "2": "spora",
15
- "3": "locky",
16
- "4": "dridex",
17
- "5": "fallout",
18
- "6": "angler",
19
- "7": "icedid",
20
- "8": "unclassified",
21
- "9": "remcos",
22
- "10": "crypt",
23
- "11": "zeus",
24
- "12": "dreambot",
25
- "13": "chthonic",
26
- "14": "tofsee",
27
- "15": "kovter",
28
- "16": "banload",
29
- "17": "reposfxg",
30
- "18": "crthrazy",
31
- "19": "troldesh",
32
- "20": "hancitor",
33
- "21": "globeimposter",
34
- "22": "rig",
35
- "23": "vawtrak",
36
- "24": "zeus-panda",
37
- "25": "neutrino",
38
- "26": "parite",
39
- "27": "miuref",
40
- "28": "sigma",
41
- "29": "upatre",
42
- "30": "ursnif",
43
- "31": "spelevo",
44
- "32": "cerber",
45
- "33": "azorult",
46
- "34": "nymaim",
47
- "35": "bunitu",
48
- "36": "ramnit",
49
- "37": "gandcrab",
50
- "38": "bankerx",
51
- "39": "trickbot",
52
- "40": "boleto",
53
- "41": "qakbot"
54
  },
55
  "initializer_range": 0.02,
56
  "intermediate_size": 256,
57
  "label2id": {
58
- "angler": 6,
59
- "azorult": 33,
60
- "bankerx": 38,
61
- "banload": 16,
62
- "boleto": 40,
63
- "bunitu": 35,
64
- "cerber": 32,
65
- "chthonic": 13,
66
- "crthrazy": 18,
67
- "crypt": 10,
68
- "dreambot": 12,
69
- "dridex": 4,
70
- "emotet": 1,
71
- "fallout": 5,
72
- "gandcrab": 37,
73
- "globeimposter": 21,
74
- "gootkit": 0,
75
- "hancitor": 20,
76
- "icedid": 7,
77
- "kovter": 15,
78
- "locky": 3,
79
- "miuref": 27,
80
- "neutrino": 25,
81
- "nymaim": 34,
82
- "parite": 26,
83
- "qakbot": 41,
84
- "ramnit": 36,
85
- "remcos": 9,
86
- "reposfxg": 17,
87
- "rig": 22,
88
- "sigma": 28,
89
- "spelevo": 31,
90
- "spora": 2,
91
- "tofsee": 14,
92
- "trickbot": 39,
93
- "troldesh": 19,
94
- "unclassified": 8,
95
- "upatre": 29,
96
- "ursnif": 30,
97
- "vawtrak": 23,
98
- "zeus": 11,
99
- "zeus-panda": 24
100
  },
101
  "layer_norm_eps": 1e-12,
102
  "max_position_embeddings": 256,
103
  "model_type": "bert",
104
  "num_attention_heads": 8,
105
  "num_hidden_layers": 24,
106
- "pad_token_id": 193,
107
  "position_embedding_type": "absolute",
108
- "torch_dtype": "float32",
109
  "transformers_version": "4.20.1",
110
  "type_vocab_size": 2,
111
  "use_cache": true,
112
- "vocab_size": 196
113
  }
 
1
  {
 
2
  "architectures": [
3
  "BertForSequenceClassification"
4
  ],
 
8
  "hidden_dropout_prob": 0.1,
9
  "hidden_size": 256,
10
  "id2label": {
11
+ "0": "trickbot",
12
+ "1": "neutrino",
13
+ "2": "banload",
14
+ "3": "crypt",
15
+ "4": "azorult",
16
+ "5": "zeus",
17
+ "6": "cerber",
18
+ "7": "locky",
19
+ "8": "gootkit",
20
+ "9": "sigma",
21
+ "10": "dridex",
22
+ "11": "emotet",
23
+ "12": "spelevo",
24
+ "13": "gandcrab",
25
+ "14": "upatre",
26
+ "15": "chthonic",
27
+ "16": "globeimposter",
28
+ "17": "ursnif",
29
+ "18": "nymaim",
30
+ "19": "angler",
31
+ "20": "vawtrak",
32
+ "21": "tofsee",
33
+ "22": "icedid",
34
+ "23": "boleto",
35
+ "24": "bankerx",
36
+ "25": "hancitor",
37
+ "26": "spora",
38
+ "27": "fallout",
39
+ "28": "crthrazy",
40
+ "29": "dreambot",
41
+ "30": "qakbot",
42
+ "31": "kovter",
43
+ "32": "unclassified",
44
+ "33": "zeus-panda",
45
+ "34": "ramnit",
46
+ "35": "troldesh",
47
+ "36": "remcos",
48
+ "37": "bunitu",
49
+ "38": "rig",
50
+ "39": "parite",
51
+ "40": "reposfxg",
52
+ "41": "miuref"
53
  },
54
  "initializer_range": 0.02,
55
  "intermediate_size": 256,
56
  "label2id": {
57
+ "angler": 19,
58
+ "azorult": 4,
59
+ "bankerx": 24,
60
+ "banload": 2,
61
+ "boleto": 23,
62
+ "bunitu": 37,
63
+ "cerber": 6,
64
+ "chthonic": 15,
65
+ "crthrazy": 28,
66
+ "crypt": 3,
67
+ "dreambot": 29,
68
+ "dridex": 10,
69
+ "emotet": 11,
70
+ "fallout": 27,
71
+ "gandcrab": 13,
72
+ "globeimposter": 16,
73
+ "gootkit": 8,
74
+ "hancitor": 25,
75
+ "icedid": 22,
76
+ "kovter": 31,
77
+ "locky": 7,
78
+ "miuref": 41,
79
+ "neutrino": 1,
80
+ "nymaim": 18,
81
+ "parite": 39,
82
+ "qakbot": 30,
83
+ "ramnit": 34,
84
+ "remcos": 36,
85
+ "reposfxg": 40,
86
+ "rig": 38,
87
+ "sigma": 9,
88
+ "spelevo": 12,
89
+ "spora": 26,
90
+ "tofsee": 21,
91
+ "trickbot": 0,
92
+ "troldesh": 35,
93
+ "unclassified": 32,
94
+ "upatre": 14,
95
+ "ursnif": 17,
96
+ "vawtrak": 20,
97
+ "zeus": 5,
98
+ "zeus-panda": 33
99
  },
100
  "layer_norm_eps": 1e-12,
101
  "max_position_embeddings": 256,
102
  "model_type": "bert",
103
  "num_attention_heads": 8,
104
  "num_hidden_layers": 24,
105
+ "pad_token_id": 234,
106
  "position_embedding_type": "absolute",
 
107
  "transformers_version": "4.20.1",
108
  "type_vocab_size": 2,
109
  "use_cache": true,
110
+ "vocab_size": 237
111
  }
tf_model.h5 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:11da57b6fe9f212ed6c053946548315f4ce52699a0be354031d1caa3b1880e22
3
- size 39320240
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7db7b0c4ec1559e03756f8d153b45be7351198b97b7913f531a6f905cb36470d
3
+ size 39362224