machine learning assignment solution