fix oobee prediction if balance classes is true
runit checks this