def predict_proba(self, X):
        """Return probability estimates for the test data X.

        Parameters
        ----------