Skip to content

Applied ML algorithms on Enron Dataset to predict Person of Interest (POI)

Notifications You must be signed in to change notification settings

mohdkashif93/Enrol-POI-Classification

Repository files navigation

Enrol-POI-Classification

I have used various Machine Learning algorithms over the Enron Dataset and analzed their performance after parameter tuning .

Install

This project requires Python 2.7 and the following Python libraries installed:

You will also need to have software installed to run and execute an iPython Notebook

We recommend students install Anaconda, a pre-packaged Python distribution that contains all of the necessary libraries and software for this project.

Run

In a terminal or command window, navigate to the top-level project directory finding_donors/ (that contains this README) and run one of the following commands:

ipython notebook Enron Email POI classification.ipynb

or

jupyter notebook Enron Email POI classification.ipynb

TODO:

  1. Add XGboost
  2. Perform Recursive feature Selection
  3. Add Graph for feature analysis
  4. Add more ensemble methods

About

Applied ML algorithms on Enron Dataset to predict Person of Interest (POI)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published