CSC3022H - Machine learning assignment