2013-05-12 87 views
0

我可以使用二元和多類標籤分類器來預測結果嗎?DecisionTreeClassifier可以同時使用二進制和多類標籤嗎?

多類標籤可以有超過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]]) 

回答

2

我認爲你是混淆標籤(通常用Y變量,如你在你的例子中做的那樣)與特徵(在你的例子中爲X)。在討論二進制或多類標籤時,通常意味着輸出/響應變量Y

輸入變量(矩陣X中的值)可以是任意整數或浮點數,所以您的示例完全有效。

+0

如果您的輸入變量代碼爲多項式值(即0 =「黑色」,1 ='紅色',2 ='綠色'),則可以先將它們二進制化,而不是直接將它們輸入分類器。 – 2013-05-13 14:43:44

相關問題