mllib:java.lang.IllegalArgumentException: GiniAggregator given label 2.0 but requires label < numCla

標籤：機器學習   spark   資料採礦   數學   scala   

報錯資訊 ：

 java.lang.IllegalArgumentException: GiniAggregator given label 2.0 but requires label < numClasses (= 2) 


在用mllib做分類，當用到一些分類演算法時，常常需要加入基尼係數。
程式碼：

RandomForest.trainClassifier(validData,2,Map[Int,Int](),10,"auto","gini",8,32)

當遇到如上報錯資訊時，注意：label<numClasses，演算法要求gini係數要求小於類的個數。此時根據報錯提示資訊將numClasses改成比你label數量更大的數字即可。

要瞭解label與numClasses對應關係的原因，還需進一步研究演算法。

mllib:java.lang.IllegalArgumentException: GiniAggregator given label 2.0 but requires label < numCla