Machine Learning Code for CS7641