Add an assert. Predicted values should be integers !!!!