Sklearn DecisionTreeClassifier如何返回几个可能的类?

时间:2019-11-28 12:02:14

标签: python machine-learning scikit-learn

我是否可以配置sklearn tree.DecisionTreeClassifier返回几个可能的类(如果进入它们的可能性相同或接近)?

我在ML方面并不擅长,因此不确定是否可行。

为了更好地理解,我尝试将其简化为简化数据:

t_X = [
    [1, 1, 0, 0],
    [1, 0, 1, 0],
    [1, 0, 0, 1]
]
t_y = [0, 1, 2]

X1 = [[0, 1, 0, 0]]的答案是[0],而且是我想要的。

X2 = [[1, 0, 0, 0]]的答案是[1],但我想要[0, 1, 2]

0 个答案:

没有答案