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... |