A Pytorch implementation of MalConv
This is the implementation of MalConv proposed in Malware Detection by Eating a Whole EXE.
Please make sure each of them is installed with the correct version
- numpy
- pytorch (0.3.0.post4)
- pandas (0.20.3)
For the training data, please place PE files under data/train/ and build the label table for training set with each row being
<File Name>, <Label>
where label = 1 refers to malware. Validation set should be handled in the same way.
Run the following command for training progress
python3 train.py <config_file_path> <random_seed>
Example : python3 train.py config/example.yaml 123
Log file, prediction on validation set & Model checkpoint will be stored at the path specified in config file.
For parameters and options availible, please refer to config/example.yaml.