mal-tls-mobilebert / config.json
SharpAI's picture
add model
0cfa022
{
"_name_or_path": "SharpAI/mal_tls-mobilebert",
"architectures": [
"BertForSequenceClassification"
],
"attention_probs_dropout_prob": 0.1,
"classifier_activation": true,
"classifier_dropout": null,
"embedding_size": 128,
"hidden_act": "relu",
"hidden_dropout_prob": 0.0,
"hidden_size": 256,
"id2label": {
"0": "gootkit",
"1": "crthrazy",
"2": "parite",
"3": "qakbot",
"4": "zeus-panda",
"5": "chthonic",
"6": "bankerx",
"7": "remcos",
"8": "sigma",
"9": "bunitu",
"10": "azorult",
"11": "spelevo",
"12": "miuref",
"13": "globeimposter",
"14": "ramnit",
"15": "dridex",
"16": "spora",
"17": "locky",
"18": "kovter",
"19": "neutrino",
"20": "nymaim",
"21": "reposfxg",
"22": "cerber",
"23": "angler",
"24": "vawtrak",
"25": "emotet",
"26": "hancitor",
"27": "crypt",
"28": "ursnif",
"29": "banload",
"30": "boleto",
"31": "gandcrab",
"32": "upatre",
"33": "fallout",
"34": "tofsee",
"35": "zeus",
"36": "trickbot",
"37": "icedid",
"38": "troldesh",
"39": "dreambot",
"40": "unclassified",
"41": "rig"
},
"initializer_range": 0.02,
"intermediate_size": 256,
"intra_bottleneck_size": 128,
"key_query_shared_bottleneck": true,
"label2id": {
"angler": 23,
"azorult": 10,
"bankerx": 6,
"banload": 29,
"boleto": 30,
"bunitu": 9,
"cerber": 22,
"chthonic": 5,
"crthrazy": 1,
"crypt": 27,
"dreambot": 39,
"dridex": 15,
"emotet": 25,
"fallout": 33,
"gandcrab": 31,
"globeimposter": 13,
"gootkit": 0,
"hancitor": 26,
"icedid": 37,
"kovter": 18,
"locky": 17,
"miuref": 12,
"neutrino": 19,
"nymaim": 20,
"parite": 2,
"qakbot": 3,
"ramnit": 14,
"remcos": 7,
"reposfxg": 21,
"rig": 41,
"sigma": 8,
"spelevo": 11,
"spora": 16,
"tofsee": 34,
"trickbot": 36,
"troldesh": 38,
"unclassified": 40,
"upatre": 32,
"ursnif": 28,
"vawtrak": 24,
"zeus": 35,
"zeus-panda": 4
},
"layer_norm_eps": 1e-12,
"max_position_embeddings": 256,
"model_type": "mobilebert",
"normalization_type": "no_norm",
"num_attention_heads": 8,
"num_feedforward_networks": 4,
"num_hidden_layers": 24,
"pad_token_id": 234,
"torch_dtype": "float32",
"transformers_version": "4.20.1",
"trigram_input": true,
"true_hidden_size": 128,
"type_vocab_size": 2,
"use_bottleneck": true,
"use_bottleneck_attention": false,
"vocab_size": 237
}