melissachall
/

cysecbert-apt-classifier

Model card Files Files and versions

cysecbert-apt-classifier / README.md

itsmelisssssaa's picture

Add APT classification model and configuration

8ffb7bf about 1 year ago

|

Raw History Blame Contribute Delete

456 Bytes

CySecBERT APT Classifier

Advanced Persistent Threat group classification using CySecBERT.

Model Details

Architecture: CySecBERTMaxPerformance
Classes: 12 APT groups
F1-Score: 95.74%
Training time: 12.6 minutes
Model size: 0.41GB

Usage

from transformers import AutoTokenizer
import torch

# Load model
model = torch.load("best_cysecbert_max_performance.pt")
tokenizer = AutoTokenizer.from_pretrained("markusbayer/CySecBERT")