分類と回帰 - RDD ベース API

spark.mllib パッケージは、2 値分類、多クラス分類、および回帰分析のさまざまなメソッドをサポートしています。以下の表は、各タイプの問題でサポートされるアルゴリズムの概要を示しています。

問題タイプ	サポートされるメソッド
2 値分類	線形 SVM、ロジスティック回帰、決定木、ランダムフォレスト、勾配ブースティング木、ナイーブベイズ
多クラス分類	ロジスティック回帰、決定木、ランダムフォレスト、ナイーブベイズ
回帰	線形最小二乗法、ラッソ、リッジ回帰、決定木、ランダムフォレスト、勾配ブースティング木、単調回帰

これらのメソッドの詳細は、こちらで確認できます。