Sklearn-如何预测所有目标标签的概率

Sklearn-如何预测所有目标标签的概率

您可以通过简单地删除

OneVsRestClassifer

和使用

predict_proba

方法来做到这一点

DecisionTreeClassifier

。您可以执行以下 *** 作：

clf = DecisionTreeClassifier()clf.fit(X_train, y_train)pred = clf.predict_proba(X_test)

这将为您提供7种可能类别中每一种的概率。

希望有帮助！