Implementation of "Multitask Learning for Network Traffic Classification". For more information read the paper.
Multitask Learning Approach for Network Traffic Classification
- Copy the entire QUIC dataset (https://drive.google.com/drive/folders/1Pvev0hJ82usPh6dWDlz7Lv8L6h3JpWhE) into the a folder, named Data.
- Run the dataProcessQuic.py to perform pre-processing.
- Now, you can run singletask.py, transferlearning.py or multitaskMasked.py. The parameters are defined at the begining of each file and they are self-explanatory.