SAGENET Pre-trained Models

Pre-trained models for SAGENET - a protocol reverse engineering tool for format inference using deep learning.

Model Overview

Model	Protocol	Size
arp_best.pth	ARP	90MB
dns_best.pth	DNS	90MB
icmp_best.pth	ICMP	90MB
mbtcp_best.pth	Modbus TCP	90MB
nbns_best.pth	NBNS	90MB
ntp_best.pth	NTP	90MB
s7comm_best.pth	S7comm	90MB
smb_best.pth	SMB	90MB
tcp_best.pth	TCP	90MB
udp_best.pth	UDP	90MB

Quick Start

from huggingface_hub import hf_hub_download

# Download a specific model
model_path = hf_hub_download(
    repo_id="Prescot12/SAGENET",
    filename="checkpoints/tcp_best.pth"
)

Usage

See the main repository for complete usage instructions.

# Test with pre-trained models
python main.py --mode test --protocols tcp udp dns

License

Apache 2.0