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

Preprocessing

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

Feature Selectors

• SelectKBest (for some functions)
• VarianceThreshold

LightGBM

• LGBMClassifier
• LGBMRanker
• LGBMRegressor

XGBoost

• XGBClassifier
• XGBRegressor

ONNX.ML

• TreeEnsembleClassifier
• TreeEnsembleRegressor