Use weighted probability in classifier