分類と回帰 - RDDベースAPI

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

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

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