ClassifierChain uses the prediction from the classifier for each label as a feature for the next label's classifier. Instead of chaining on the categorical prediction it should be able to select a method to chain on, one of { predict, predict_proba, decision_function }.