machine-learning assignment 4