二叉决策树分类器的class MulticlassClassificationEvaluator

时间:2020-03-24 14:37:45

标签: apache-spark-ml

我想使用 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)  

0 个答案:

没有答案