2014-08-12 20:15:21 -04:00
|
|
|
---
|
|
|
|
layout: global
|
|
|
|
title: Classification and Regression - MLlib
|
|
|
|
displayTitle: <a href="mllib-guide.html">MLlib</a> - Classification and Regression
|
|
|
|
---
|
|
|
|
|
|
|
|
MLlib supports various methods for
|
|
|
|
[binary classification](http://en.wikipedia.org/wiki/Binary_classification),
|
|
|
|
[multiclass
|
|
|
|
classification](http://en.wikipedia.org/wiki/Multiclass_classification), and
|
|
|
|
[regression analysis](http://en.wikipedia.org/wiki/Regression_analysis). The table below outlines
|
|
|
|
the supported algorithms for each type of problem.
|
|
|
|
|
|
|
|
<table class="table">
|
|
|
|
<thead>
|
|
|
|
<tr><th>Problem Type</th><th>Supported Methods</th></tr>
|
|
|
|
</thead>
|
|
|
|
<tbody>
|
|
|
|
<tr>
|
2015-02-25 19:13:17 -05:00
|
|
|
<td>Binary Classification</td><td>linear SVMs, logistic regression, decision trees, random forests, gradient-boosted trees, naive Bayes</td>
|
2014-08-12 20:15:21 -04:00
|
|
|
</tr>
|
|
|
|
<tr>
|
2015-02-25 19:13:17 -05:00
|
|
|
<td>Multiclass Classification</td><td>decision trees, random forests, naive Bayes</td>
|
2014-08-12 20:15:21 -04:00
|
|
|
</tr>
|
|
|
|
<tr>
|
2015-02-25 19:13:17 -05:00
|
|
|
<td>Regression</td><td>linear least squares, Lasso, ridge regression, decision trees, random forests, gradient-boosted trees, isotonic regression</td>
|
2014-08-12 20:15:21 -04:00
|
|
|
</tr>
|
|
|
|
</tbody>
|
|
|
|
</table>
|
|
|
|
|
|
|
|
More details for these methods can be found here:
|
|
|
|
|
|
|
|
* [Linear models](mllib-linear-methods.html)
|
|
|
|
* [binary classification (SVMs, logistic regression)](mllib-linear-methods.html#binary-classification)
|
|
|
|
* [linear regression (least squares, Lasso, ridge)](mllib-linear-methods.html#linear-least-squares-lasso-and-ridge-regression)
|
|
|
|
* [Decision trees](mllib-decision-tree.html)
|
2015-02-25 19:13:17 -05:00
|
|
|
* [Ensembles of decision trees](mllib-ensembles.html)
|
|
|
|
* [random forests](mllib-ensembles.html#random-forests)
|
|
|
|
* [gradient-boosted trees](mllib-ensembles.html#gradient-boosted-trees-gbts)
|
2014-08-12 20:15:21 -04:00
|
|
|
* [Naive Bayes](mllib-naive-bayes.html)
|
2015-02-15 12:10:03 -05:00
|
|
|
* [Isotonic regression](mllib-isotonic-regression.html)
|