cwe-predictor / config.json
mulliken's picture
Upload DistilBertForSequenceClassification
5ceac98 verified
{
"activation": "gelu",
"architectures": [
"DistilBertForSequenceClassification"
],
"attention_dropout": 0.1,
"dim": 768,
"dropout": 0.1,
"hidden_dim": 3072,
"id2label": {
"0": "CWE-1",
"1": "CWE-1021",
"2": "CWE-113",
"3": "CWE-116",
"4": "CWE-117",
"5": "CWE-118",
"6": "CWE-1187",
"7": "CWE-1188",
"8": "CWE-119",
"9": "CWE-120",
"10": "CWE-121",
"11": "CWE-122",
"12": "CWE-123",
"13": "CWE-1236",
"14": "CWE-125",
"15": "CWE-126",
"16": "CWE-1284",
"17": "CWE-1286",
"18": "CWE-129",
"19": "CWE-131",
"20": "CWE-1321",
"21": "CWE-1333",
"22": "CWE-134",
"23": "CWE-16",
"24": "CWE-17",
"25": "CWE-170",
"26": "CWE-172",
"27": "CWE-178",
"28": "CWE-18",
"29": "CWE-184",
"30": "CWE-185",
"31": "CWE-189",
"32": "CWE-19",
"33": "CWE-190",
"34": "CWE-191",
"35": "CWE-193",
"36": "CWE-199",
"37": "CWE-20",
"38": "CWE-200",
"39": "CWE-201",
"40": "CWE-202",
"41": "CWE-203",
"42": "CWE-208",
"43": "CWE-209",
"44": "CWE-21",
"45": "CWE-212",
"46": "CWE-214",
"47": "CWE-22",
"48": "CWE-23",
"49": "CWE-240",
"50": "CWE-248",
"51": "CWE-250",
"52": "CWE-252",
"53": "CWE-254",
"54": "CWE-255",
"55": "CWE-256",
"56": "CWE-259",
"57": "CWE-264",
"58": "CWE-266",
"59": "CWE-269",
"60": "CWE-270",
"61": "CWE-273",
"62": "CWE-275",
"63": "CWE-276",
"64": "CWE-281",
"65": "CWE-283",
"66": "CWE-284",
"67": "CWE-285",
"68": "CWE-287",
"69": "CWE-288",
"70": "CWE-290",
"71": "CWE-294",
"72": "CWE-295",
"73": "CWE-297",
"74": "CWE-300",
"75": "CWE-303",
"76": "CWE-305",
"77": "CWE-306",
"78": "CWE-307",
"79": "CWE-310",
"80": "CWE-311",
"81": "CWE-312",
"82": "CWE-319",
"83": "CWE-320",
"84": "CWE-321",
"85": "CWE-326",
"86": "CWE-327",
"87": "CWE-330",
"88": "CWE-331",
"89": "CWE-332",
"90": "CWE-335",
"91": "CWE-338",
"92": "CWE-342",
"93": "CWE-345",
"94": "CWE-346",
"95": "CWE-347",
"96": "CWE-349",
"97": "CWE-35",
"98": "CWE-350",
"99": "CWE-352",
"100": "CWE-353",
"101": "CWE-354",
"102": "CWE-358",
"103": "CWE-359",
"104": "CWE-36",
"105": "CWE-361",
"106": "CWE-362",
"107": "CWE-367",
"108": "CWE-369",
"109": "CWE-377",
"110": "CWE-378",
"111": "CWE-379",
"112": "CWE-384",
"113": "CWE-385",
"114": "CWE-388",
"115": "CWE-399",
"116": "CWE-400",
"117": "CWE-401",
"118": "CWE-404",
"119": "CWE-405",
"120": "CWE-407",
"121": "CWE-415",
"122": "CWE-416",
"123": "CWE-417",
"124": "CWE-425",
"125": "CWE-426",
"126": "CWE-427",
"127": "CWE-428",
"128": "CWE-434",
"129": "CWE-436",
"130": "CWE-441",
"131": "CWE-444",
"132": "CWE-457",
"133": "CWE-459",
"134": "CWE-470",
"135": "CWE-471",
"136": "CWE-472",
"137": "CWE-476",
"138": "CWE-489",
"139": "CWE-494",
"140": "CWE-497",
"141": "CWE-502",
"142": "CWE-521",
"143": "CWE-522",
"144": "CWE-532",
"145": "CWE-534",
"146": "CWE-538",
"147": "CWE-548",
"148": "CWE-552",
"149": "CWE-565",
"150": "CWE-59",
"151": "CWE-601",
"152": "CWE-61",
"153": "CWE-610",
"154": "CWE-611",
"155": "CWE-613",
"156": "CWE-617",
"157": "CWE-639",
"158": "CWE-640",
"159": "CWE-642",
"160": "CWE-644",
"161": "CWE-657",
"162": "CWE-662",
"163": "CWE-664",
"164": "CWE-665",
"165": "CWE-667",
"166": "CWE-668",
"167": "CWE-669",
"168": "CWE-670",
"169": "CWE-672",
"170": "CWE-674",
"171": "CWE-680",
"172": "CWE-681",
"173": "CWE-682",
"174": "CWE-693",
"175": "CWE-697",
"176": "CWE-704",
"177": "CWE-706",
"178": "CWE-707",
"179": "CWE-73",
"180": "CWE-732",
"181": "CWE-74",
"182": "CWE-749",
"183": "CWE-754",
"184": "CWE-755",
"185": "CWE-757",
"186": "CWE-763",
"187": "CWE-769",
"188": "CWE-77",
"189": "CWE-770",
"190": "CWE-772",
"191": "CWE-776",
"192": "CWE-78",
"193": "CWE-787",
"194": "CWE-788",
"195": "CWE-789",
"196": "CWE-79",
"197": "CWE-798",
"198": "CWE-799",
"199": "CWE-80",
"200": "CWE-805",
"201": "CWE-807",
"202": "CWE-822",
"203": "CWE-823",
"204": "CWE-824",
"205": "CWE-829",
"206": "CWE-834",
"207": "CWE-835",
"208": "CWE-838",
"209": "CWE-840",
"210": "CWE-843",
"211": "CWE-862",
"212": "CWE-863",
"213": "CWE-88",
"214": "CWE-89",
"215": "CWE-90",
"216": "CWE-908",
"217": "CWE-909",
"218": "CWE-91",
"219": "CWE-912",
"220": "CWE-913",
"221": "CWE-915",
"222": "CWE-916",
"223": "CWE-917",
"224": "CWE-918",
"225": "CWE-920",
"226": "CWE-922",
"227": "CWE-924",
"228": "CWE-93",
"229": "CWE-94",
"230": "CWE-943",
"231": "CWE-99"
},
"initializer_range": 0.02,
"label2id": {
"CWE-1": 0,
"CWE-1021": 1,
"CWE-113": 2,
"CWE-116": 3,
"CWE-117": 4,
"CWE-118": 5,
"CWE-1187": 6,
"CWE-1188": 7,
"CWE-119": 8,
"CWE-120": 9,
"CWE-121": 10,
"CWE-122": 11,
"CWE-123": 12,
"CWE-1236": 13,
"CWE-125": 14,
"CWE-126": 15,
"CWE-1284": 16,
"CWE-1286": 17,
"CWE-129": 18,
"CWE-131": 19,
"CWE-1321": 20,
"CWE-1333": 21,
"CWE-134": 22,
"CWE-16": 23,
"CWE-17": 24,
"CWE-170": 25,
"CWE-172": 26,
"CWE-178": 27,
"CWE-18": 28,
"CWE-184": 29,
"CWE-185": 30,
"CWE-189": 31,
"CWE-19": 32,
"CWE-190": 33,
"CWE-191": 34,
"CWE-193": 35,
"CWE-199": 36,
"CWE-20": 37,
"CWE-200": 38,
"CWE-201": 39,
"CWE-202": 40,
"CWE-203": 41,
"CWE-208": 42,
"CWE-209": 43,
"CWE-21": 44,
"CWE-212": 45,
"CWE-214": 46,
"CWE-22": 47,
"CWE-23": 48,
"CWE-240": 49,
"CWE-248": 50,
"CWE-250": 51,
"CWE-252": 52,
"CWE-254": 53,
"CWE-255": 54,
"CWE-256": 55,
"CWE-259": 56,
"CWE-264": 57,
"CWE-266": 58,
"CWE-269": 59,
"CWE-270": 60,
"CWE-273": 61,
"CWE-275": 62,
"CWE-276": 63,
"CWE-281": 64,
"CWE-283": 65,
"CWE-284": 66,
"CWE-285": 67,
"CWE-287": 68,
"CWE-288": 69,
"CWE-290": 70,
"CWE-294": 71,
"CWE-295": 72,
"CWE-297": 73,
"CWE-300": 74,
"CWE-303": 75,
"CWE-305": 76,
"CWE-306": 77,
"CWE-307": 78,
"CWE-310": 79,
"CWE-311": 80,
"CWE-312": 81,
"CWE-319": 82,
"CWE-320": 83,
"CWE-321": 84,
"CWE-326": 85,
"CWE-327": 86,
"CWE-330": 87,
"CWE-331": 88,
"CWE-332": 89,
"CWE-335": 90,
"CWE-338": 91,
"CWE-342": 92,
"CWE-345": 93,
"CWE-346": 94,
"CWE-347": 95,
"CWE-349": 96,
"CWE-35": 97,
"CWE-350": 98,
"CWE-352": 99,
"CWE-353": 100,
"CWE-354": 101,
"CWE-358": 102,
"CWE-359": 103,
"CWE-36": 104,
"CWE-361": 105,
"CWE-362": 106,
"CWE-367": 107,
"CWE-369": 108,
"CWE-377": 109,
"CWE-378": 110,
"CWE-379": 111,
"CWE-384": 112,
"CWE-385": 113,
"CWE-388": 114,
"CWE-399": 115,
"CWE-400": 116,
"CWE-401": 117,
"CWE-404": 118,
"CWE-405": 119,
"CWE-407": 120,
"CWE-415": 121,
"CWE-416": 122,
"CWE-417": 123,
"CWE-425": 124,
"CWE-426": 125,
"CWE-427": 126,
"CWE-428": 127,
"CWE-434": 128,
"CWE-436": 129,
"CWE-441": 130,
"CWE-444": 131,
"CWE-457": 132,
"CWE-459": 133,
"CWE-470": 134,
"CWE-471": 135,
"CWE-472": 136,
"CWE-476": 137,
"CWE-489": 138,
"CWE-494": 139,
"CWE-497": 140,
"CWE-502": 141,
"CWE-521": 142,
"CWE-522": 143,
"CWE-532": 144,
"CWE-534": 145,
"CWE-538": 146,
"CWE-548": 147,
"CWE-552": 148,
"CWE-565": 149,
"CWE-59": 150,
"CWE-601": 151,
"CWE-61": 152,
"CWE-610": 153,
"CWE-611": 154,
"CWE-613": 155,
"CWE-617": 156,
"CWE-639": 157,
"CWE-640": 158,
"CWE-642": 159,
"CWE-644": 160,
"CWE-657": 161,
"CWE-662": 162,
"CWE-664": 163,
"CWE-665": 164,
"CWE-667": 165,
"CWE-668": 166,
"CWE-669": 167,
"CWE-670": 168,
"CWE-672": 169,
"CWE-674": 170,
"CWE-680": 171,
"CWE-681": 172,
"CWE-682": 173,
"CWE-693": 174,
"CWE-697": 175,
"CWE-704": 176,
"CWE-706": 177,
"CWE-707": 178,
"CWE-73": 179,
"CWE-732": 180,
"CWE-74": 181,
"CWE-749": 182,
"CWE-754": 183,
"CWE-755": 184,
"CWE-757": 185,
"CWE-763": 186,
"CWE-769": 187,
"CWE-77": 188,
"CWE-770": 189,
"CWE-772": 190,
"CWE-776": 191,
"CWE-78": 192,
"CWE-787": 193,
"CWE-788": 194,
"CWE-789": 195,
"CWE-79": 196,
"CWE-798": 197,
"CWE-799": 198,
"CWE-80": 199,
"CWE-805": 200,
"CWE-807": 201,
"CWE-822": 202,
"CWE-823": 203,
"CWE-824": 204,
"CWE-829": 205,
"CWE-834": 206,
"CWE-835": 207,
"CWE-838": 208,
"CWE-840": 209,
"CWE-843": 210,
"CWE-862": 211,
"CWE-863": 212,
"CWE-88": 213,
"CWE-89": 214,
"CWE-90": 215,
"CWE-908": 216,
"CWE-909": 217,
"CWE-91": 218,
"CWE-912": 219,
"CWE-913": 220,
"CWE-915": 221,
"CWE-916": 222,
"CWE-917": 223,
"CWE-918": 224,
"CWE-920": 225,
"CWE-922": 226,
"CWE-924": 227,
"CWE-93": 228,
"CWE-94": 229,
"CWE-943": 230,
"CWE-99": 231
},
"max_position_embeddings": 512,
"model_type": "distilbert",
"n_heads": 12,
"n_layers": 6,
"pad_token_id": 0,
"problem_type": "single_label_classification",
"qa_dropout": 0.1,
"seq_classif_dropout": 0.2,
"sinusoidal_pos_embds": false,
"tie_weights_": true,
"torch_dtype": "float32",
"transformers_version": "4.55.0",
"vocab_size": 30522
}