AutoSKLearn是否有SKLearn中predict_proba的等效方法？

AutoSKLearn完全兼容scikit-learn的API，其中就包含与predict_proba等效的方法，方法名称也叫predict_proba，专门用于获取分类任务中每个样本属于各类别的概率值（也就是你要的预测置信度）。

基本用法示例 #

当你用AutoSKLearn训练完分类模型后，直接调用该方法即可：

import autosklearn.classification

# 初始化并训练模型
automl = autosklearn.classification.AutoSklearnClassifier()
automl.fit(X_train, y_train)

# 获取预测概率（置信度）
probabilities = automl.predict_proba(X_test)

返回的probabilities是一个二维数组，每行对应一个测试样本，每列对应一个类别的预测概率，所有类别概率之和为1。

注意事项 #

• 该方法仅适用于分类任务，回归任务没有对应的概率输出，因为回归是预测连续值。
• 如果你的模型是在多分类场景下训练的，predict_proba会返回每个样本在所有类别上的概率分布；二分类场景下则返回每个样本属于正类和负类的概率。

内容的提问来源于stack exchange，提问作者John Westlund