/cesnet-models

CESNET Models: Neural networks for network traffic classification

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

The goal of this project is to provide neural network architectures for traffic classification and their pre-trained weights.

The newest network architecture is named Multi-modal CESNET v2 (mm-CESNET-v2) and is visualized in the following picture. See the getting started page and models reference for more information.

🐸 🐸 See a related project CESNET DataZoo providing large TLS and QUIC traffic datasets. 🐸 🐸

📓 📓 Example Jupyter notebooks are included in a separate CESNET Traffic Classification Examples repo. 📓 📓

Multi-modal CESNET v2

Installation

Install the package from pip with:

pip install cesnet-models

or for editable install with:

pip install -e git+https://github.com/CESNET/cesnet-models

Papers

Models from the following papers are included: