分類と回帰 - RDDベースAPI
spark.mllibパッケージは、二項分類、多クラス分類、および回帰分析のさまざまなメソッドをサポートしています。以下の表は、各問題タイプでサポートされているアルゴリズムの概要です。
| 問題タイプ | サポートされているメソッド |
|---|---|
| 二項分類 | 線形SVM、ロジスティック回帰、決定木、ランダムフォレスト、勾配ブースティング木、ナイーブベイズ |
| 多クラス分類 | ロジスティック回帰、決定木、ランダムフォレスト、ナイーブベイズ |
| 回帰 | 線形最小二乗法、Lasso、リッジ回帰、決定木、ランダムフォレスト、勾配ブースティング木、単調回帰 |
これらのメソッドの詳細はこちらで確認できます。