Allow predictor method to be defined in the model definition.
HACK: This is necessary becuase predict_proba returns 2 columns. What's a general solution here?