machine learning assignment