itsmelisssssaa commited on
Commit
8ffb7bf
·
1 Parent(s): dc0932e

Add APT classification model and configuration

Browse files
Files changed (3) hide show
  1. README.md +19 -0
  2. best_cysecbert_max_performance.pt +3 -0
  3. config.json +9 -0
README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # CySecBERT APT Classifier
2
+
3
+ Advanced Persistent Threat group classification using CySecBERT.
4
+
5
+ ## Model Details
6
+ - Architecture: CySecBERTMaxPerformance
7
+ - Classes: 12 APT groups
8
+ - F1-Score: 95.74%
9
+ - Training time: 12.6 minutes
10
+ - Model size: 0.41GB
11
+
12
+ ## Usage
13
+ ```python
14
+ from transformers import AutoTokenizer
15
+ import torch
16
+
17
+ # Load model
18
+ model = torch.load("best_cysecbert_max_performance.pt")
19
+ tokenizer = AutoTokenizer.from_pretrained("markusbayer/CySecBERT")
best_cysecbert_max_performance.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c32ef320c2133d03982b6bb7606685f27c17d91009d6d0559f4250cc221bbc09
3
+ size 440146673
config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "cysecbert-apt",
3
+ "architectures": ["CySecBERTMaxPerformance"],
4
+ "num_classes": 12,
5
+ "hidden_size": 768,
6
+ "max_position_embeddings": 512,
7
+ "model_class": "APTClassifier",
8
+ "framework": "pytorch"
9
+ }