Machine learning assignment