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

• MLPClassifier
• BernoulliNB
• GaussianNB
• MultinomialNB

Preprocessing

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

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
• Linear Classifier/Regressor
• Normalizer
• Scaler
• TreeEnsembleClassifier
• TreeEnsembleRegressor