PB Add classification feature. Add partition depending on trainig percentage