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 Item-based collaborative filtering Alternating Least Squares (ALS) |
|
Feature extraction |
Feature extraction and transformation |
TF-IDF Word2Vec Standard Scaler Normalizer Chi-Square Selector |
|
Optimization |
Optimization |
Stochastic Gradient Descent Limited... |
