Skip to content

Supported Models & Transformers

Akshay Utkarsh Sharma edited this page Feb 8, 2018 · 5 revisions

Spark Mllib

  • LogisticRegressionModel
  • DecisionTreeModel
  • RandomForestModel
  • GradientBoostedTreesModel
  • StringIndexer
  • HashingTF
  • OneHotEncoder
  • RegexTokenizer
  • CountVectorizer
  • StandardScaler
  • MinMaxScaler
  • Bucketizer
  • VectorAssembler
  • ChiSqFeatureSelector

Custom Models

  • Log1PScaler
  • Algebraic transformer
  • Probabilistic transformer

Pipelines

  • Pipeline using any of supported models
Clone this wiki locally