PML project: class prediction