Класс MulticlassClassificationEvaluator для двоичного классификатора дерева решений - PullRequest
Новая
       21

Класс MulticlassClassificationEvaluator для двоичного классификатора дерева решений

0 голосов
/ 24 марта 2020

Я хочу использовать класс MulticlassClassificationEvaluator , чтобы использовать метрическую c "точность" с использованием перекрестной проверки с бинарным деревом решений. Мне было интересно, если производительность этого класса () MulticlassClassificationEvaluator ) было бы нормально, используя двоичное дерево решений (2 класса)? Вместо этого следует использовать BinaryClassificationEvaluator , поскольку он не имеет метрики точности c

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)
...