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

时间：2015-04-09 21:53:48 阅读：2337 评论：0 收藏：0 [点我收藏+]

报错信息：

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

踩

(0)

评论一句话评论（0）

分享档案

更多>