我是否可以配置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]