在 sklearn 中,有没有一种方法可以测试对我的 DecisionTreeClassifier 模型做出的预测的置信度?

Mar*_*ane 3 python machine-learning scipy scikit-learn

我成功地在 python 中使用了 sklearn 库并且非常享受它。

我能够使用以下代码创建和拟合 DecisionTreeClassifierType 的模型:

clf = tree.DecisionTreeClassifier()

clf = clf.fit(features, labels)
Run Code Online (Sandbox Code Playgroud)

然后我可以使用该模型来预测新输入的类别,如下所示:

clf.predict([[20, 50, 10]])
Run Code Online (Sandbox Code Playgroud)

上面的行将返回 0 或 1,具体取决于模型预测此数据将具有的类别。我想知道是否有某种方法可以获得模型对预测的置信度/概率?

因此,如果它预测输入的分类为 1,则概率/置信度将是小数点,如 0.8 或百分比,如 80%。关于这是否与 sklearn 的 DecisionTreeClassifier 兼容/可能的任何想法?

max*_*moo 5

这是在sklearn.tree.DecisionTreeClassifier.predict_proba

预测输入样本 X 的类别概率。预测类别概率是叶子中同一类别样本的比例。