Supported Operators

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