Machine learning algorithms

The following table provides a list of algorithms supported by MLlib with classifications such as the type of machine learning and the type of algorithm:

Type of machine learning

Type of algorithm

Algorithm name

Supervised learning


Naive Bayes

Decision Trees

Random Forests

Gradient-Boosted Trees


Linear Regression

Logistic Regression

Support Vector Machines

Unsupervised learning



Gaussian mixture

Power Iteration Clustering (PIC)

Latent Dirichlet Allocation (LDA)

Streaming k-means

Dimensionality reduction

Singular Value Decomposition (SVD)

Principal Component Analysis (PCA)

Recommender systems

Collaborative filtering

User-based collaborative filtering

Get Big Data Analytics now with O’Reilly online learning.

O’Reilly members experience live online training, plus books, videos, and digital content from 200+ publishers.