如何在决策树中使用python调出每个类的概率值

Question

我该如何从预测模型中找出每个类别值的概率。

Python代码：

clf = DecisionTreeClassifier(criterion='gini',
    splitter='best',
    max_depth=None,
    min_samples_split=2,
    min_samples_leaf=1,
    min_weight_fraction_leaf=0.0,
    max_features=None,
    random_state=123,
    max_leaf_nodes=None,
    min_impurity_decrease=0.0,
    min_impurity_split=None,
    class_weight=None,
    presort=False)

# Train Decision Tree Classifer
clf = clf.fit(X_train,y_train)

# Predict the response for test dataset
y_pred = clf.predict(X_test)

＃查看具有预测值的数据集    dtree_gini = pd.DataFrame（{'Actual'：y_test，'Predicted'：y_pred}）    dtree_gini1 = pd.merge（X_test，dtree_gini，left_index = True，right_index = True）;    dtree_gini1.head（5）

data

desire outcomes

谢谢！

Answer 1

您可以像这样使用函数predict_proba：

y_proba = clf.predict_proba(X_test)