分类和回归 - 基于RDD的API

这个 spark.mllib 包支持多种方法用于 二分类 , 多分类 ,以及 回归分析 。下面的表格列出了每种问题类型支持的算法。

问题类型 支持的方法
二元分类 线性支持向量机, 逻辑回归, 决策树, 随机森林, 梯度提升树, 朴素贝叶斯
多类分类 逻辑回归, 决策树, 随机森林, 朴素贝叶斯
回归 线性最小二乘法, Lasso, 岭回归, 决策树, 随机森林, 梯度提升树, 单调回归

有关这些方法的更多细节可以在这里找到: