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