machine learning assignment 1