Suggest allowing predictions to be both int or float

Its a bit silly when the model I use outputs integers and then needing to change them to float.