'Predict the target of new samples. Can be different from the
prediction of the uncalibrated classifier.
Parameters
X : array-like, shape (n_samples, n_features)
The samples.
Returns
C : array, shape (n_samples,)
The predicted class.'

def predict(self, X):

    check_is_fitted(self, ['classes_', 'calibrated_classifiers_'])
    return self.classes_[np.argmax(self.predict_proba(X), axis=1)]
