Full production source code and trained PyTorch model for an AI-powered network intrusion detection system, trained on the real-world CIC-IDS2017 dataset (~3M flows) — not a toy demo. FastAPI backend, GPU-accelerated training pipeline, ONNX export.
Complete backend codebase: FastAPI + PyTorch
Advanced training pipeline — AMP mixed precision, LR range finder, cosine annealing, early stopping
Production-ready inference function with scaler, confusion-matrix generation, ONNX export