@arjoly I turned this into a decision_function and staged_decision_function that supports binary and mutli-class for both real and discrete versions of the algorithm.