Antivirus Demo for Fresh Machine Learning #7
Switch branches/tags
Nothing to show
Clone or download
Latest commit 60090b6 Sep 8, 2016
Permalink
Failed to load latest commit information.
classifier first Aug 13, 2016
.gitignore first Aug 13, 2016
LICENSE first Aug 13, 2016
README.md Update README.md Sep 8, 2016
checkmanalyzer.py first Aug 13, 2016
checkpe.py first Aug 13, 2016
data.csv first Aug 13, 2016
generatedata.py first Aug 13, 2016
learning.py first Aug 13, 2016

README.md

Antivirus Demo

Overview

This project helps train a classifier to be able to detect PE files as either malicious or legitimate. It tries out 6 different classification algorithms before deciding which one to use for prediction by comparing their results. This is the code for 'Build an Antivirus in 5 Min' on Youtube.

Dependencies

  • pandas pip install pandas
  • numpy pip install numpy
  • pickle pip install pickle
  • scipy pip install scipy
  • scikit pip install -U scikit-learn

Use pip to install any missing dependencies

Basic Usage

  1. Run python learning.py to train the model. It will train on the dataset included called 'data.csv'.

  2. Once trained you can test the model via python checkpe.py YOUR_PE_FILE. It will output either malicious or legitimate!

That's it!

Credits

Credit for the vast majority of code here goes to Te-k. I've merely created a wrapper around all of the important functions to get people started.