Skip to content

Dhayson/WGAN-intrusion-detection

Repository files navigation

Como rodar:

Treinar modelos

Self Attention

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 train sa

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 train sa

Self Attention + WGAN-GP

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 train sa-gp

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 train sa-gp

TCN

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 train tcn

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 train tcn

TCN + WGAN-GP

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 train tcn-gp

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 train tcn-gp

LSTM

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 train lstm

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 train lstm

Executar modelos (após o treinamento)

Serão criados arquivos out_curve_[conjunto].png e out_matrix_[conjunto].png, onde [conjunto] será val ou test dependendo do que for escolhido. Outras métricas serão printadas no terminal.

Self Attention

Dataset CIC-DDoS-2019

Conjunto de validação
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 val thresh both sa
Conjunto de teste
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 test thresh both sa

Dataset CIC-IDS-2017

Conjunto de validação
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 val thresh both sa
Conjunto de teste
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 test thresh both sa

Self Attention + WGAN-GP

Dataset CIC-DDoS-2019

Conjunto de validação
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 val thresh both sa-gp
Conjunto de teste
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 test thresh both sa-gp

Dataset CIC-IDS-2017

Conjunto de validação
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 val thresh both sa-gp
Conjunto de teste
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 test thresh both sa-gp

TCN

Dataset CIC-DDoS-2019

Conjunto de validação
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 val thresh both tcn
Conjunto de teste
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 test thresh both tcn

Dataset CIC-IDS-2017

Conjunto de validação
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 val thresh both tcn
Conjunto de teste
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 test thresh both tcn

TCN + WGAN-GP

Dataset CIC-DDoS-2019

Conjunto de validação
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 val thresh both tcn-gp
Conjunto de teste
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 test thresh both tcn-gp

Dataset CIC-IDS-2017

Conjunto de validação
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 val thresh both tcn-gp
Conjunto de teste
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 test thresh both tcn-gp

LSTM

Dataset CIC-DDoS-2019

Conjunto de validação
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 val thresh both lstm
Conjunto de teste
python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 test thresh both lstm

Dataset CIC-IDS-2017

Conjunto de validação
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 val thresh both lstm
Conjunto de teste
python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 test thresh both lstm

Realizar tunagem de parâmetros

Self Attention

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 tune sa

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 tune sa

Self Attention + WGAN-GP

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 tune sa-gp

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 tune sa-gp

TCN

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 optuna tcn

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 optuna tcn

TCN + WGAN-GP

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 optuna tcn-gp

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 optuna tcn-gp

LSTM

Dataset CIC-DDoS-2019

python3 main.py 2019 dataset_filtered/Day1 dataset_filtered/Day2 optuna lstm

Dataset CIC-IDS-2017

python3 main.py 2017 dataset_alternative_filtered/Day1 dataset_alternative_filtered/Day3 dataset_alternative_filtered/Day5 optuna lstm

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages