Spark ML,参数" rawPredictionCol"用于二进制分类

时间:2016-05-05 16:56:25

标签: scala apache-spark apache-spark-ml

我想在Spark.ml中使用二进制分类器来在我的管道之后评估我的模型。我使用这段代码:

val gbt = new GBTClassifier()
  .setLabelCol("Label_Index")
  .setFeaturesCol("features")
  .setMaxIter(10)
  .setMaxDepth(7)
  .setSubsamplingRate(0.1)
  .setMinInstancesPerNode(15)


// Convert indexed labels back to original labels.
  val labelConverter = new IndexToString()
  .setInputCol("prediction")
  .setOutputCol("predictedLabel")
  .setLabels(indexer_2.labels)

val evaluator_auc = new BinaryClassificationEvaluator()
.setLabelCol("Label_Index")
.setRawPredictionCol("")
.setMetricName("areaUnderROC")

我真的不知道我需要给哪些参数" setRawPredictionCol()",我想我需要给出我的预测结果,列#34;预测&# 34;

0 个答案:

没有答案