from sklearn.linear_model import LogisticRegression from sklearn.calibration import CalibratedClassifierCV # base classifier clf = LogisticRegression() # sigmoid calibration calibrated_clf = CalibratedClassifierCV(clf, cv=5, method='sigmoid') # train calibrated classifier calibrated_clf.fit(X_train, y_train) # predict probabilities y_prob = calibrated_clf.predict_proba(X_test)
from sklearn.ensemble import RandomForestClassifier from sklearn.calibration import CalibratedClassifierCV # base classifier clf = RandomForestClassifier() # isotonic calibration calibrated_clf = CalibratedClassifierCV(clf, cv=5, method='isotonic') # train calibrated classifier calibrated_clf.fit(X_train, y_train) # predict probabilities y_prob = calibrated_clf.predict_proba(X_test)In both examples, the `CalibratedClassifierCV` is used to improve the accuracy of the predicted probabilities of the base classifier (`LogisticRegression` in example 1 and `RandomForestClassifier` in example 2) by fitting a probability model (`sigmoid` and `isotonic`, respectively) to the outputs of the base classifier. The `CalibratedClassifierCV` is a part of the scikit-learn package library in Python, which is used for machine learning and data analysis purposes.