Machine Learning Final Project (CS 4774 UVA)