Modified code for the Udacity/GA Tech Machine Learning course final project.