There might be use-cases where you can predict a probability instead of just a label. We should support that.
There might be use-cases where you can predict a probability instead of just a label. We should support that.