fixing binomial classification prediction