Machine Learning Engineer Nanodegree

Capstone Project

The Capstone is a two-staged project. The first is the proposal component, where you can receive valuable feedback about your project idea, design, and proposed solution. This must be completed prior to your implementation and submitting for the capstone project.

You can find the capstone proposal and the capstone project in the project directory.

All data required to build the model is located inside the directory 'data'.

This project uses an employee attrition dataset to develop a prediction model that assess the probability of an employee leave the company given certain features.

In order to develop the model the following libraries were used:

• Numpy
• Pandas
• Scikit Learn
• Matplotlib

All development was done in Jupiter Notebook, the file is located in the directory under the name "Capstone".