This is a PyTorch/GPU implementation of the paper Ensemble Model Based on an Improved Convolutional Neural:
@Article{EnsembleModelBasedOnAnImprovedConvolutionalNeural2022,
Add the final citation
}
- Download NSL-KDD and pretrains
- Train
- Test and ensemble to pretrain models
Download the NSL-KDD dataset on your local, run code bellow on project root
./download.sh
After that, you will see new ./Data directory on project root.
python train.py --dir ./Data
./download_pretrains.sh
Now there will be existed ./Pretrains directory.
python eval.py --pretrain ./Pretrains