EtiTarun commited on
Commit
a478e49
·
verified ·
1 Parent(s): 5de3535

Pushed best fine-tuned CodeBERT binary vuln classifier

Browse files
Files changed (2) hide show
  1. config.json +5 -63
  2. model.safetensors +2 -2
config.json CHANGED
@@ -10,72 +10,14 @@
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
- "0": "CWE111",
14
- "1": "CWE113",
15
- "2": "CWE121",
16
- "3": "CWE122",
17
- "4": "CWE124",
18
- "5": "CWE126",
19
- "6": "CWE127",
20
- "7": "CWE129",
21
- "8": "CWE134",
22
- "9": "CWE190",
23
- "10": "CWE191",
24
- "11": "CWE194",
25
- "12": "CWE195",
26
- "13": "CWE197",
27
- "14": "CWE23",
28
- "15": "CWE36",
29
- "16": "CWE369",
30
- "17": "CWE400",
31
- "18": "CWE401",
32
- "19": "CWE415",
33
- "20": "CWE457",
34
- "21": "CWE486",
35
- "22": "CWE590",
36
- "23": "CWE690",
37
- "24": "CWE762",
38
- "25": "CWE78",
39
- "26": "CWE789",
40
- "27": "CWE80",
41
- "28": "CWE89",
42
- "29": "safe",
43
- "30": "vulnerable"
44
  },
45
  "initializer_range": 0.02,
46
  "intermediate_size": 3072,
47
  "label2id": {
48
- "CWE111": 0,
49
- "CWE113": 1,
50
- "CWE121": 2,
51
- "CWE122": 3,
52
- "CWE124": 4,
53
- "CWE126": 5,
54
- "CWE127": 6,
55
- "CWE129": 7,
56
- "CWE134": 8,
57
- "CWE190": 9,
58
- "CWE191": 10,
59
- "CWE194": 11,
60
- "CWE195": 12,
61
- "CWE197": 13,
62
- "CWE23": 14,
63
- "CWE36": 15,
64
- "CWE369": 16,
65
- "CWE400": 17,
66
- "CWE401": 18,
67
- "CWE415": 19,
68
- "CWE457": 20,
69
- "CWE486": 21,
70
- "CWE590": 22,
71
- "CWE690": 23,
72
- "CWE762": 24,
73
- "CWE78": 25,
74
- "CWE789": 26,
75
- "CWE80": 27,
76
- "CWE89": 28,
77
- "safe": 29,
78
- "vulnerable": 30
79
  },
80
  "layer_norm_eps": 1e-05,
81
  "max_position_embeddings": 514,
@@ -86,7 +28,7 @@
86
  "pad_token_id": 1,
87
  "position_embedding_type": "absolute",
88
  "torch_dtype": "float32",
89
- "transformers_version": "4.55.0",
90
  "type_vocab_size": 1,
91
  "use_cache": true,
92
  "vocab_size": 50265
 
10
  "hidden_dropout_prob": 0.1,
11
  "hidden_size": 768,
12
  "id2label": {
13
+ "0": "0",
14
+ "1": "1"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
15
  },
16
  "initializer_range": 0.02,
17
  "intermediate_size": 3072,
18
  "label2id": {
19
+ "0": 0,
20
+ "1": 1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
21
  },
22
  "layer_norm_eps": 1e-05,
23
  "max_position_embeddings": 514,
 
28
  "pad_token_id": 1,
29
  "position_embedding_type": "absolute",
30
  "torch_dtype": "float32",
31
+ "transformers_version": "4.52.4",
32
  "type_vocab_size": 1,
33
  "use_cache": true,
34
  "vocab_size": 50265
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:427aec4295bb985391efa418564ce8a097e6090bba99c85d5e7f6b025ff5e31e
3
- size 498702028
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:862cb6d3d47d6882952abdea6efe0fef2f6a62d238780f8047498684b8aaec3c
3
+ size 498612824