machine learning assignment ans..