什么是 LogisticRegressionWithLBFGS Spark-Mllib 中的 setNumClasses

Question

我不明白 setNumClasses 的意义是什么，在 sparkmllib 文档中也找不到任何内容。

new LogisticRegressionWithLBFGS()
      .setNumClasses(10)

Answer 1

setNumClasses 方法设置多项逻辑回归中 k 类 分类问题的可能结果数。

在这种情况下，您可以设置 k = 10。

Answer 2

Link 到文档

https://spark.apache.org/docs/latest/api/scala/index.html#org.apache.spark.mllib.classification.LogisticRegressionWithLBFGS

def
setNumClasses(numClasses: Int): LogisticRegressionWithLBFGS.this.type

设置多项逻辑回归中第 k 类 个分类问题的可能结果数。

Answer 3

是的，您可以设置用于训练模型的不同 类 的数量，您的数据可以使用这些标签。默认值为 2，您还可以通过在训练时指定更大的数字，在调用 .train(numClasses=3) 或调用上面提到的函数时指定