Supported Operators

scikit-learn

Tree-based operators

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

Linear methods

• LinearRegression
• LinearSVC
• LogisticRegression
• LogisticRegressionCV
• SGDClassifier

SVM

• NuSVC
• SVC

Classifiers: Other

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

Preprocessing

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

Matrix Decomposition

• PCA
• KernelPCA
• TruncatedSVD
• FastICA

Feature Selectors

• SelectPercentile
• SelectKBest
• VarianceThreshold

Feature Pre-processing: One-to-One

• SimpleImputer
• MissingIndicator

Feature Pre-processing: Other

• PolynomialFeatures

LightGBM

• LGBMClassifier
• LGBMRanker
• LGBMRegressor

XGBoost

• XGBClassifier
• XBGRanker
• XGBRegressor

ONNX.ML

• ArrayFeatureExtractor
• FeatureVectorizer
• LabelEncoder (ints only for now)
• LinearClassifier
• LinearRegressor
• Normalizer
• Scaler
• TreeEnsembleClassifier
• TreeEnsembleRegressor

SparkML

• Bucketizer
• VectorAssembler
• LogisticRegressionModel