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

报错信息 :

 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对应关系的原因,还需进一步研究算法。







文章来自:http://blog.csdn.net/u013361361/article/details/44965547
© 2021 jiaocheng.bubufx.com  联系我们
ICP备案:鲁ICP备09046678号-3