什么是LogisticRegressionWithLBFGS Spark-Mllib中的setNumClasses

Question

我无法理解setNumClasses的重要性在这里也无法在sparkmllib文档中找到任何内容。

new LogisticRegressionWithLBFGS()
      .setNumClasses(10)

Answer 1

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

在这种情况下，你会设置k = 10.

Answer 2

链接到文档

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

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

在Multiomial Logistic回归中设置k类分类问题的可能结果数。

Answer 3

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