def METHOD_NAME(self, X):
    self.reload()
    if self.params["ml_task"] == BINARY_CLASSIFICATION:
        return self.model.predict_proba(X)[:, 1]
    elif self.params["ml_task"] == MULTICLASS_CLASSIFICATION:
        return self.model.predict_proba(X)
    return self.model.METHOD_NAME(X)