我想使用 MulticlassClassificationEvaluator 类,以便将“交叉验证”与二进制决策树一起使用度量“准确性”,我想知道此类的性能( MulticlassClassificationEvaluator < / strong>)可以使用二进制决策树(2个类)吗?代替使用 BinaryClassificationEvaluator ,因为它没有准确性指标
val evaluator = new MulticlassClassificationEvaluator()
evaluator.setLabelCol("label")
evaluator.setPredictionCol("prediction")
evaluator.setMetricName("accuracy")
val cs = new CrossValidator()
.setEstimator(new DecisionTreeClassifier())
.setEvaluator(evaluator)
.setEstimatorParamMaps(paramGrid)
.setNumFolds(10)