Deep Learning-Based Encrypted Traffic Classification: Novel Use of the Headers
-
This is a Pytorch implementation of 1D-CNN Deep Packet.
-
See
107research_poster.pdf
for details.
Dataset
-
This research uses the UNB ISCX VPN-nonVPN dataset.
-
Make sure to download and unzip the dataset. Then run the following command for data preprocess:
python3 data_preprocess.py {path/to/unzipped/dataset}
-
The preprocessed dataset is saved to
./pre
.
Run
-
Make sure you have run the previous steps so that the preprocessed dataset is in
./pre
folder. -
To train the model and see the results, simply run:
python3 main.py