---

layout: global title: “MLlib: RDD-based API” displayTitle: “MLlib: RDD-based API” license: | Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements. See the NOTICE file distributed with this work for additional information regarding copyright ownership. The ASF licenses this file to You under the Apache License, Version 2.0 (the “License”); you may not use this file except in compliance with the License. You may obtain a copy of the License at

 http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an “AS IS” BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.

This page documents sections of the MLlib guide for the RDD-based API (the spark.mllib package). Please see the MLlib Main Guide for the DataFrame-based API (the spark.ml package), which is now the primary API for MLlib.

• Data types
• Basic statistics
  • summary statistics
  • correlations
  • stratified sampling
  • hypothesis testing
  • streaming significance testing
  • random data generation
• Classification and regression
  • linear models (SVMs, logistic regression, linear regression)
  • naive Bayes
  • decision trees
  • ensembles of trees (Random Forests and Gradient-Boosted Trees)
  • isotonic regression
• Collaborative filtering
  • alternating least squares (ALS)
• Clustering
  • k-means
  • Gaussian mixture
  • power iteration clustering (PIC)
  • latent Dirichlet allocation (LDA)
  • bisecting k-means
  • streaming k-means
• Dimensionality reduction
  • singular value decomposition (SVD)
  • principal component analysis (PCA)
• Feature extraction and transformation
• Frequent pattern mining
  • FP-growth
  • association rules
  • PrefixSpan
• Evaluation metrics
• PMML model export
• Optimization (developer)
  • stochastic gradient descent
  • limited-memory BFGS (L-BFGS)