Machine learning midterm solution