Supported Transformers

Here is a list of all supported transformers in all of the core platforms that MLeap supports.

NOTE: Tensorflow has no listed support here, but it is possible toinclude Tensorflow graphs in an MLeap transformer pipeline.

Features

Transformer Spark MLeap Scikit-Learn TensorFlow
Binarizer x x x
BucketedRandomProjectionLSH x x
Bucketizer x x
ChiSqSelector x x
CountVectorizer x x
DCT x x
ElementwiseProduct x x x
HashingTermFrequency x x x
IDF x x
Imputer x x x
Interaction x x x
MaxAbsScaler x x
MinHashLSH x x
MinMaxScaler x x x
Ngram x x
Normalizer x x
OneHotEncoder x x x
PCA x x x
QuantileDiscretizer x x
PolynomialExpansion x x x
ReverseStringIndexer x x x
StandardScaler x x x
StopWordsRemover x x
StringIndexer x x x
Tokenizer x x x
VectorAssembler x x x
VectorIndexer x x
VectorSlicer x x
WordToVector x x

Classification

Transformer Spark MLeap Scikit-Learn TensorFlow
DecisionTreeClassifier x x x
GradientBoostedTreeClassifier x x
LogisticRegression x x x
LogisticRegressionCv x x x
NaiveBayesClassifier x x
OneVsRest x x
RandomForestClassifier x x x
SupportVectorMachines x x x
MultiLayerPerceptron x x

Regression

Transformer Spark MLeap Scikit-Learn TensorFlow
AFTSurvivalRegression x x
DecisionTreeRegression x x x
GeneralizedLinearRegression x x
GradientBoostedTreeRegression x x
IsotonicRegression x x
LinearRegression x x x
RandomForestRegression x x x

Clustering

Transformer Spark MLeap Scikit-Learn TensorFlow
BisectingKMeans x x
GaussianMixtureModel x x
KMeans x x
LDA x

Extensions

Transformer Spark MLeap Scikit-Learn TensorFlow Description
MathUnary x x x Simple set of unary mathematical operations
MathBinary x x x Simple set of binary mathematical operations
StringMap x x x Maps a string to a double

Recommendation

Transformer Spark MLeap Scikit-Learn TensorFlow
ALS x