DecisionTreeClassifier 如何找到下一个值

Question

从训练集中构建决策树后如何找到下一个值？ 在我的示例中，当我输入 B 和 C 值时，我需要知道 A 字段的下一个值的概率。

df = pd.DataFrame(np.random.randint(low=0, high=10, size=(5, 3)),
                    columns=['a', 'b', 'c'])

>>>
    a   b   c
0   2   8   8
1   4   2   9
2   1   0   7
3   5   1   7
4   6   0   2

X = df[ ['b','c'] ]
Y = df[ ['a'] ]

clf = DecisionTreeClassifier(random_state=241)
clf.fit(X, Y)

如果 B=8, C=8，如何找出字段 A 的下一个值的概率？

Answer 1

您可以简单地使用clf.predict_proba()。例如：

clf.predict_proba([[8,8]])

将返回一个浮点向量，表示数据是“a”列中每个唯一值的概率。