Skip to content

Alexander-H-Liu/MalConv-Pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

939cb59 · Jan 20, 2018

History

7 Commits
Jan 20, 2018
Jan 20, 2018
Jan 20, 2018
Jan 20, 2018
Jan 20, 2018
Jan 20, 2018
Jan 20, 2018
Jan 20, 2018
Jan 20, 2018

Repository files navigation

MalConv-Pytorch

A Pytorch implementation of MalConv


Desciprtion

This is the implementation of MalConv proposed in Malware Detection by Eating a Whole EXE.

Dependency

Please make sure each of them is installed with the correct version

  • numpy
  • pytorch (0.3.0.post4)
  • pandas (0.20.3)

Setup

Preparing data

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.

Training

Run the following command for training progress

    python3 train.py <config_file_path> <random_seed>
    Example : python3 train.py config/example.yaml 123

Training Log & Checkpoint

Log file, prediction on validation set & Model checkpoint will be stored at the path specified in config file.

Parameters & Model Options

For parameters and options availible, please refer to config/example.yaml.

About

Pytorch implementation of MalConv

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages