DecisionTreeClassifier可以同时与二进制和多类标签一起使用吗?

时间:2013-05-12 13:35:23

标签: python scikit-learn

我可以使用二元和多类标签来拟合分类器来预测结果吗?

多类标签可以有2个以上的值,二进制标签只能有2个。

示例(X中的第一个参数是多类,第二个参数是二进制 - [-1,1]):

from sklearn import tree
X = [[0, -1], [2, 1], [1, -1] ]
Y = [0, 1, 1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(X, Y)
clf.predict([[1, 1]])

1 个答案:

答案 0 :(得分:2)

我认为您使用功能(示例中的矩阵Y)会混淆标签(通常用X变量表示,就像您在示例中所做的那样)。在谈论二进制或多类标签时,通常意味着输出/响应变量Y

输入变量(矩阵X中的值)可以是任意整数或浮点数,因此您的示例完全有效。