mohdkashif93/Enrol-POI-Classification

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

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:

• NumPy
• Pandas
• matplotlib
• scikit-learn

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