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

Classification

Naive Bayes

Decision Trees

Random Forests

Gradient-Boosted Trees

Regression

Linear Regression

Logistic Regression

Support Vector Machines

Unsupervised learning

Clustering

K-Means

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 the O’Reilly learning platform.

O’Reilly members experience books, live events, courses curated by job role, and more from O’Reilly and nearly 200 top publishers.