Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
ArjitJ authored Jun 4, 2021
1 parent 0b982b9 commit 4d3abcc
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ MemStream augments a representation learning algorithm like PCA, IB and AutoEnco
5. SYN: Run `python3 memstream-syn.py --dataset SYN --beta 1 --memlen 16`

## Command line options
* `--dataset`: The dataset to be used for training. Choices 'NSL', 'KDD', 'UNSW', 'DOS' and 'IDS'. (default 'NSL')
* `--dataset`: The dataset to be used for training. Choices 'NSL', 'KDD', 'UNSW', 'DOS'. (default 'NSL')
* `--beta`: The threshold beta to be used. (default: 1e-3)
* `--dim`: The dimensionality of the representations learnt by the representation learning algorithm (default: 12)
* `--memlen`: The size of the Memory Module (default: 512)
Expand All @@ -27,8 +27,7 @@ MemStream expects the input multi-aspect record stream to be stored in a contain
2. [UNSW-NB 15](https://www.unsw.adfa.edu.au/unsw-canberra-cyber/cybersecurity/ADFA-NB15-Datasets/)
3. [KDDCUP99](http://kdd.ics.uci.edu/databases/kddcup99/kddcup99.html)
4. [NSL-KDD](https://www.unb.ca/cic/datasets/nsl.html)
5. [ISCX2012](https://www.unb.ca/cic/datasets/ids.html) -- Not included in the folder due to space limitation
6. Synthetic Dataset
5. Synthetic Dataset

## Environment
This code has been tested on Debian GNU/Linux 9 with a 12GB Nvidia GeForce RTX 2080 Ti GPU, CUDA Version 10.2 and PyTorch 1.5.
This code has been tested on Debian GNU/Linux 9 with a 12GB Nvidia GeForce RTX 2080 Ti GPU, CUDA Version 10.2 and PyTorch 1.5.

0 comments on commit 4d3abcc

Please sign in to comment.