Skip to content

Supported Operators

Matteo Interlandi edited this page Jan 25, 2023 · 41 revisions

scikit-learn

Tree-based operators

  • DecisionTreeClassifier
  • DecisionTreeRegressor
  • ExtraTreesClassifier
  • ExtraTreesRegressor
  • GradientBoostingClassifier
  • GradientBoostingRegressor
  • HistGradientBoostingClassifier
  • HistGradientBoostingRegressor
  • IsolationForest
  • RandomForestClassifier
  • RandomForestRegressor

Linear methods

  • LinearRegression
  • LogisticRegression
  • LogisticRegressionCV
  • RidgeCV
  • Ridge
  • SGDClassifier
  • Lasso
  • ElasticNet

SVM

  • LinearSVC
  • LinearSVR
  • NuSVC
  • SVC

Clustering

  • KMeans
  • MeanShift

Classifiers: Other

  • BernoulliNB
  • GaussianNB
  • KNeighborsClassifier
  • KNeighborsRegressor
  • MLPClassifier
  • MLPRegressor
  • MultinomialNB

Preprocessing

  • Binarizer
  • LabelEncoder
  • Normalizer
  • OneHotEncoder
  • RobustScaler
  • MaxAbsScaler
  • MinMaxScaler
  • StandardScaler

Matrix Decomposition

  • PCA
  • KernelPCA
  • TruncatedSVD
  • FastICA

Cross Decomposition

  • PLS Regression

Feature Selectors

  • SelectPercentile
  • SelectKBest
  • VarianceThreshold

Feature Pre-processing: One-to-One

  • SimpleImputer
  • MissingIndicator

Feature Pre-processing: Other

  • PolynomialFeatures

Mixture

  • BayesianGaussianMixture

Other

  • FeatureUnion
  • GridSearchCV
  • ColumnTransformer
  • Pipeline
  • RandomizedGridSearchCV
  • StackingClassifier
  • StackingRegressor
  • PLSRegressor

LightGBM

  • Booster
  • LGBMClassifier
  • LGBMRanker
  • LGBMRegressor

XGBoost

  • XGBClassifier
  • XBGRanker
  • XGBRegressor

ONNX.ML

  • Abs
  • Add
  • ArrayFeatureExtractor
  • Binarizer
  • Cast
  • Concat
  • Div
  • Imputer
  • LabelEncoder
  • Less
  • LinearClassifier
  • LinearRegressor
  • Mul
  • Neg
  • Normalizer
  • OneHotEncoder
  • Reshape
  • Sum
  • Scaler
  • SVMClassifier
  • TreeEnsembleClassifier
  • TreeEnsembleRegressor

SparkML

  • Bucketizer
  • VectorAssembler
  • LogisticRegressionModel

Prophet

  • Trend prediction