MalwareClassify

Setup

In order to execute our project a few packages are required. To install them please run: pip install -r requirements.txt

Execute the script

usage: classification_main.py [-h] -f FILE [-d PDS_DB] [-m MODEL_WEIGHTS]
                              [-p PCA] [-s SCALER] [--pds-only] [--model-only]

Malware Classifier

optional arguments:
  -h, --help            show this help message and exit
  -f FILE, --file FILE  Path to the ember sample to be examined
  -d PDS_DB, --pds-db PDS_DB
                        Path to the PDS database
  -m MODEL_WEIGHTS, --model-weights MODEL_WEIGHTS
                        Path to the weights of the model
  -p PCA, --pca PCA     Path to the pca object
  -s SCALER, --scaler SCALER
                        Path to the scaler object
  --pds-only            Classify by using only the PDS filter
  --model-only          Classify by using only the NN filter

To execute the script on a Windows PE file please run: python classification_main.py -f /path/to/pe/

Note: make sure the model_weights.h5, pca.pkl and scaler.pkl are in the same folder that your run from or provide thier path to the command line arguments.