python - 使用 Python API 进行逻辑回归多类分类

Question

目前，Python API 尚不支持 Spark 内的多类分类，但将来会支持，如 Spark 页面 1 中所述。 。

是否有任何发布日期或是否有机会使用 Python 运行它，通过 Logistic 回归实现多类？我知道它可以用 Scala 运行，但我想用 Python 运行它。谢谢。

Answer 1

scikit-learn 的 LogisticRegression提供一个 multi_class 参数。来自文档:

Multiclass option can be either ‘ovr’ or ‘multinomial’. If the option chosen is ‘ovr’, then a binary problem is fit for each label. Else the loss minimised is the multinomial loss fit across the entire probability distribution. Works only for the ‘lbfgs’ solver.

因此，multi_class='ovr' 似乎是您的正确选择。

了解更多信息:see this link

<小时/>

添加:

根据 pyspark 文档，您仍然可以使用其 API 进行多类回归。使用类pyspark.mllib.classification.LogisticRegressionWithLBFGS，您可以获得用于多类分类的可选参数numClasses。