可以使用Apache Spark的支持向量机进行多类分类吗?

当我尝试使用Apache Spark SVM解决多类分类问题时,遇到了以下错误。有人能解释一下是否有办法使用Apache Spark MLlib进行SVM多类分类吗?

Exception in thread "main" org.apache.spark.SparkException: Input validation failed.    at org.apache.spark.mllib.regression.GeneralizedLinearAlgorithm.run(GeneralizedLinearAlgorithm.scala:251)    at org.apache.spark.mllib.regression.GeneralizedLinearAlgorithm.run(GeneralizedLinearAlgorithm.scala:229)    at org.apache.spark.mllib.classification.SVMWithSGD$.train(SVM.scala:219)    at org.apache.spark.mllib.classification.SVMWithSGD$.train(SVM.scala:255)    at org.apache.spark.mllib.classification.SVMWithSGD.train(SVM.scala)    at SVMClass.main(SVMClass.java:31)    at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)    at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)    at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)    at java.lang.reflect.Method.invoke(Method.java:498)    at com.intellij.rt.execution.application.AppMain.main(AppMain.java:144)

回答:

并非所有机器学习算法都能直接处理多类问题。如果是这种情况,你总是可以使用一对多策略。维基百科文章解释了哪些算法具有“自然”的多类扩展能力。

如果你查看Spark文档,你会注意到SVM被列在二元分类下。SVM算法需要扩展才能处理多个类别,显然在MLlib中可能尚未实现(从文档中判断)。你可以通过提到的OvR策略来解决这个问题,但性能可能不会非常出色。多层感知器是一个有趣的替代方案,因为它也可以提供属于给定类的概率。

Related Posts

L1-L2正则化的不同系数

我想对网络的权重同时应用L1和L2正则化。然而,我找不…

使用scikit-learn的无监督方法将列表分类成不同组别,有没有办法?

我有一系列实例,每个实例都有一份列表,代表它所遵循的不…

f1_score metric in lightgbm

我想使用自定义指标f1_score来训练一个lgb模型…

通过相关系数矩阵进行特征选择

我在测试不同的算法时,如逻辑回归、高斯朴素贝叶斯、随机…

可以将机器学习库用于流式输入和输出吗?

已关闭。此问题需要更加聚焦。目前不接受回答。 想要改进…

在TensorFlow中,queue.dequeue_up_to()方法的用途是什么?

我对这个方法感到非常困惑,特别是当我发现这个令人费解的…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注