Views
No views yet
| Head | Task |
|---|---|
cls_head | Malicious / benign (log_classification, phishing) |
mitre_head | MITRE technique classification (791 classes) |
ner_head | IOC token NER (17 BIO labels) |
anomaly_head | Anomaly score regression |
1from training.model import AstraeaConfig, AstraeaModel
2import torch
3
4cfg = AstraeaConfig(
5 base_model_name="answerdotai/ModernBERT-base",
6 num_mitre_classes=791,
7)
8model = AstraeaModel(cfg)
9state = torch.load("pytorch_model.bin", map_location="cpu", weights_only=True)
10model.load_state_dict(state, strict=False)1{
2 "model_id": "lunaris-sentinel",
3 "base_model": "answerdotai/ModernBERT-base",
4 "checkpoint_root": "/root/neobert-astraea-finetune/checkpoints/astraea-modernbert-full",
5 "primary_weights": "/root/neobert-astraea-finetune/checkpoints/astraea-modernbert-full/checkpoint-3000",
6 "num_mitre_classes": 791,
7 "num_ner_labels": 17,
8 "checkpoints_included": [
9 "checkpoint-2000",
10 "checkpoint-2500",
11 "checkpoint-3000"
12 ]
13}