Supported Converters

The following scikit-learn estimators and transformers have a registered converter in yobx.sklearn. The list is generated programmatically from the live converter registry. External-library estimators from lightgbm, xgboost, category_encoders, and imbalanced-learn are listed when the corresponding optional dependencies are installed; see External Libraries Based on scikit-learn for architecture details.

Coverage Table

The table below lists all scikit-learn estimators and transformers, showing which ones have a native converter in yobx.sklearn among those which can (predictable). External-library estimators are appended at the end.

category_encoders

<<<

from yobx.sklearn import register_sklearn_converters
from yobx.sklearn.register import get_sklearn_estimator_coverage

register_sklearn_converters()

print(get_sklearn_estimator_coverage(libraries=("category_encoders",), rst=True))

>>>

category	estimator	predictable	yobx	converter	since
category_encoders	BackwardDifferenceEncoder	✓
category_encoders	BaseNEncoder	✓
category_encoders	BinaryEncoder	✓	✓	category_encoders_binary_encoder
category_encoders	CatBoostEncoder	✓
category_encoders	CountEncoder	✓
category_encoders	GLMMEncoder	✓
category_encoders	GrayEncoder	✓
category_encoders	HashingEncoder	✓
category_encoders	HelmertEncoder	✓
category_encoders	JamesSteinEncoder	✓
category_encoders	LeaveOneOutEncoder	✓
category_encoders	MEstimateEncoder	✓
category_encoders	OneHotEncoder	✓	✓	category_encoders_one_hot_encoder
category_encoders	OrdinalEncoder	✓	✓	category_encoders_ordinal_encoder
category_encoders	PolynomialEncoder	✓	✓	category_encoders_polynomial_encoder
category_encoders	QuantileEncoder	✓	✓	category_encoders_quantile_encoder
category_encoders	RankHotEncoder	✓
category_encoders	SumEncoder	✓
category_encoders	TargetEncoder	✓	✓	category_encoders_target_encoder
category_encoders	WOEEncoder	✓	✓	category_encoders_woe_encoder

Coverage: 7/20 ~ 35.0%

imbalanced-learn

<<<

from yobx.sklearn import register_sklearn_converters
from yobx.sklearn.register import get_sklearn_estimator_coverage

register_sklearn_converters()

print(get_sklearn_estimator_coverage(libraries=("imblearn",), rst=True))

>>>

category	estimator	predictable	yobx	converter	since
ensemble	BalancedBaggingClassifier	✓	✓	sklearn_balanced_bagging_classifier
ensemble	BalancedRandomForestClassifier	✓	✓	sklearn_balanced_random_forest_classifier
ensemble	EasyEnsembleClassifier	✓	✓	sklearn_easy_ensemble_classifier
ensemble	RUSBoostClassifier	✓	✓	sklearn_rus_boost_classifier

Coverage: 4/4 ~ 100.0%

lightgbm

<<<

from yobx.sklearn import register_sklearn_converters
from yobx.sklearn.register import get_sklearn_estimator_coverage

register_sklearn_converters()

print(get_sklearn_estimator_coverage(libraries=("lightgbm",), rst=True))

>>>

category	estimator	predictable	yobx	converter	since
lightgbm	LGBMClassifier	✓	✓	sklearn_lgbm_classifier
lightgbm	LGBMRanker	✓	✓	sklearn_lgbm_ranker
lightgbm	LGBMRegressor	✓	✓	sklearn_lgbm_regressor

Coverage: 3/3 ~ 100.0%

scikit-learn

<<<

from yobx.sklearn import register_sklearn_converters
from yobx.sklearn.register import get_sklearn_estimator_coverage

register_sklearn_converters()

print(get_sklearn_estimator_coverage(libraries=("sklearn",), rst=True))

>>>

category	estimator	predictable	yobx	converter	since
cluster	AffinityPropagation	✓	✓	sklearn_affinity_propagation
cluster	AgglomerativeClustering
cluster	Birch	✓	✓	sklearn_birch
cluster	BisectingKMeans	✓	✓	sklearn_bisecting_kmeans
cluster	DBSCAN
cluster	FeatureAgglomeration	✓	✓	sklearn_feature_agglomeration
cluster	KMeans	✓	✓	sklearn_kmeans
cluster	MeanShift	✓	✓	sklearn_mean_shift
cluster	MiniBatchKMeans	✓	✓	sklearn_mini_batch_kmeans
cluster	OPTICS
cluster	SpectralBiclustering
cluster	SpectralClustering
cluster	SpectralCoclustering
compose	ColumnTransformer	✓	✓	sklearn_column_transformer
compose	TransformedTargetRegressor	✓
covariance	EllipticEnvelope	✓	✓	sklearn_elliptic_envelope
covariance	EmpiricalCovariance	✓	✓	sklearn_empirical_covariance
covariance	GraphicalLasso	✓	✓	sklearn_graphical_lasso
covariance	GraphicalLassoCV	✓	✓	sklearn_graphical_lasso
covariance	LedoitWolf	✓	✓	sklearn_ledoit_wolf
covariance	MinCovDet	✓	✓	sklearn_min_cov_det
covariance	OAS	✓	✓	sklearn_oas
covariance	ShrunkCovariance	✓	✓	sklearn_shrunk_covariance
cross_decomposition	CCA	✓	✓	sklearn_cca
cross_decomposition	PLSCanonical	✓
cross_decomposition	PLSRegression	✓	✓	sklearn_pls_regression
cross_decomposition	PLSSVD	✓	✓	sklearn_pls_svd
decomposition	DictionaryLearning	✓
decomposition	FactorAnalysis	✓	✓	sklearn_factor_analysis
decomposition	FastICA	✓	✓	sklearn_fast_ica
decomposition	IncrementalPCA	✓	✓	sklearn_incremental_pca
decomposition	KernelPCA	✓	✓	sklearn_kernel_pca
decomposition	LatentDirichletAllocation	✓	✓	sklearn_latent_dirichlet_allocation
decomposition	MiniBatchDictionaryLearning	✓
decomposition	MiniBatchNMF	✓	✓	sklearn_nmf
decomposition	MiniBatchSparsePCA	✓	✓	sklearn_sparse_pca
decomposition	NMF	✓	✓	sklearn_nmf
decomposition	PCA	✓	✓	sklearn_pca
decomposition	SparseCoder	✓
decomposition	SparsePCA	✓	✓	sklearn_sparse_pca
decomposition	TruncatedSVD	✓	✓	sklearn_truncated_svd
ensemble	AdaBoostClassifier	✓	✓	sklearn_adaboost_classifier
ensemble	AdaBoostRegressor	✓	✓	sklearn_adaboost_regressor
ensemble	BaggingClassifier	✓	✓	sklearn_bagging_classifier
ensemble	BaggingRegressor	✓	✓	sklearn_bagging_regressor
ensemble	ExtraTreesClassifier	✓	✓	sklearn_extra_trees_classifier
ensemble	ExtraTreesRegressor	✓	✓	sklearn_extra_trees_regressor
ensemble	GradientBoostingClassifier	✓	✓	sklearn_gradient_boosting_classifier
ensemble	GradientBoostingRegressor	✓	✓	sklearn_gradient_boosting_regressor
ensemble	IsolationForest	✓	✓	sklearn_isolation_forest
ensemble	RandomForestClassifier	✓	✓	sklearn_random_forest_classifier
ensemble	RandomForestRegressor	✓	✓	sklearn_random_forest_regressor
ensemble	RandomTreesEmbedding	✓
ensemble	StackingClassifier	✓	✓	sklearn_stacking_classifier
ensemble	StackingRegressor	✓	✓	sklearn_stacking_regressor
ensemble	VotingClassifier	✓	✓	sklearn_voting_classifier
ensemble	VotingRegressor	✓	✓	sklearn_voting_regressor
feature_extraction	CountVectorizer	✓	✓	sklearn_count_vectorizer
feature_extraction	DictVectorizer	✓
feature_extraction	FeatureHasher	✓	✓	sklearn_feature_hasher
feature_extraction	HashingVectorizer	✓
feature_extraction	PatchExtractor	✓	✓	sklearn_patch_extractor
feature_extraction	TfidfTransformer	✓	✓	sklearn_tfidf_transformer
feature_extraction	TfidfVectorizer	✓	✓	sklearn_tfidf_vectorizer
feature_selection	GenericUnivariateSelect	✓
feature_selection	RFE	✓	✓	sklearn_rfe
feature_selection	RFECV	✓	✓	sklearn_rfecv
feature_selection	SelectFdr	✓	✓	sklearn_select_fdr
feature_selection	SelectFpr	✓	✓	sklearn_select_fpr
feature_selection	SelectFromModel	✓	✓	sklearn_select_from_model
feature_selection	SelectFwe	✓	✓	sklearn_select_fwe
feature_selection	SelectKBest	✓	✓	sklearn_select_k_best
feature_selection	SelectPercentile	✓	✓	sklearn_select_percentile
feature_selection	SequentialFeatureSelector	✓
feature_selection	VarianceThreshold	✓	✓	sklearn_variance_threshold
frozen	FrozenEstimator
gaussian_process	GaussianProcessClassifier	✓	✓	sklearn_gaussian_process_classifier
gaussian_process	GaussianProcessRegressor	✓	✓	sklearn_gaussian_process_regressor
glm	GammaRegressor	✓	✓	sklearn_glm_regressor
glm	PoissonRegressor	✓	✓	sklearn_glm_regressor
glm	TweedieRegressor	✓	✓	sklearn_glm_regressor
hdbscan	HDBSCAN
hist_gradient_boosting	HistGradientBoostingClassifier	✓	✓	sklearn_hgb_classifier
hist_gradient_boosting	HistGradientBoostingRegressor	✓	✓	sklearn_hgb_regressor
impute	KNNImputer	✓	✓	sklearn_knn_imputer
impute	MissingIndicator	✓	✓	sklearn_missing_indicator
impute	SimpleImputer	✓	✓	sklearn_simple_imputer
linear_model	ARDRegression	✓	✓	sklearn_linear_regressor
linear_model	BayesianRidge	✓	✓	sklearn_linear_regressor
linear_model	ElasticNet	✓	✓	sklearn_linear_regressor
linear_model	ElasticNetCV	✓	✓	sklearn_linear_regressor
linear_model	HuberRegressor	✓	✓	sklearn_linear_regressor
linear_model	Lars	✓	✓	sklearn_linear_regressor
linear_model	LarsCV	✓	✓	sklearn_linear_regressor
linear_model	Lasso	✓	✓	sklearn_linear_regressor
linear_model	LassoCV	✓	✓	sklearn_linear_regressor
linear_model	LassoLars	✓	✓	sklearn_linear_regressor
linear_model	LassoLarsCV	✓	✓	sklearn_linear_regressor
linear_model	LassoLarsIC	✓	✓	sklearn_linear_regressor
linear_model	LinearRegression	✓	✓	sklearn_linear_regressor
linear_model	LogisticRegression	✓	✓	sklearn_logistic_regression
linear_model	LogisticRegressionCV	✓	✓	sklearn_logistic_regression
linear_model	MultiTaskElasticNet	✓	✓	sklearn_linear_regressor
linear_model	MultiTaskElasticNetCV	✓	✓	sklearn_linear_regressor
linear_model	MultiTaskLasso	✓	✓	sklearn_linear_regressor
linear_model	MultiTaskLassoCV	✓	✓	sklearn_linear_regressor
linear_model	OrthogonalMatchingPursuit	✓	✓	sklearn_linear_regressor
linear_model	OrthogonalMatchingPursuitCV	✓	✓	sklearn_linear_regressor
linear_model	PassiveAggressiveClassifier	✓	✓	sklearn_linear_classifier
linear_model	PassiveAggressiveRegressor	✓	✓	sklearn_linear_regressor
linear_model	Perceptron	✓	✓	sklearn_linear_classifier
linear_model	QuantileRegressor	✓	✓	sklearn_linear_regressor
linear_model	RANSACRegressor	✓	✓	sklearn_ransac_regressor
linear_model	Ridge	✓	✓	sklearn_linear_regressor
linear_model	RidgeCV	✓	✓	sklearn_linear_regressor
linear_model	RidgeClassifier	✓	✓	sklearn_linear_classifier
linear_model	RidgeClassifierCV	✓	✓	sklearn_linear_classifier
linear_model	SGDClassifier	✓	✓	sklearn_linear_classifier
linear_model	SGDOneClassSVM	✓	✓	sklearn_sgd_one_class_svm
linear_model	SGDRegressor	✓	✓	sklearn_linear_regressor
linear_model	TheilSenRegressor	✓	✓	sklearn_linear_regressor
manifold	ClassicalMDS
manifold	Isomap	✓	✓	sklearn_isomap
manifold	LocallyLinearEmbedding	✓	✓	sklearn_locally_linear_embedding
manifold	MDS
manifold	SpectralEmbedding
manifold	TSNE
mixture	BayesianGaussianMixture	✓	✓	sklearn_bayesian_gaussian_mixture
mixture	GaussianMixture	✓	✓	sklearn_gaussian_mixture
model_selection	FixedThresholdClassifier	✓
model_selection	GridSearchCV	✓	✓	sklearn_grid_search_cv
model_selection	HalvingGridSearchCV	✓	✓	sklearn_halving_grid_search_cv
model_selection	HalvingRandomSearchCV	✓	✓	sklearn_halving_random_search_cv
model_selection	RandomizedSearchCV	✓	✓	sklearn_randomized_search_cv
model_selection	TunedThresholdClassifierCV	✓	✓	sklearn_tuned_threshold_classifier_cv	1.5+
neighbors	KNeighborsClassifier	✓	✓	sklearn_knn_classifier
neighbors	KNeighborsRegressor	✓	✓	sklearn_knn_regressor
neighbors	KNeighborsTransformer	✓	✓	sklearn_kneighbors_transformer
neighbors	KernelDensity	✓	✓	sklearn_kernel_density
neighbors	LocalOutlierFactor	✓	✓	sklearn_local_outlier_factor
neighbors	NearestCentroid	✓	✓	sklearn_nearest_centroid	1.8+
neighbors	NearestNeighbors
neighbors	NeighborhoodComponentsAnalysis	✓	✓	sklearn_neighborhood_components_analysis
neighbors	RadiusNeighborsClassifier	✓	✓	sklearn_radius_neighbors_classifier
neighbors	RadiusNeighborsRegressor	✓	✓	sklearn_radius_neighbors_regressor
neighbors	RadiusNeighborsTransformer	✓	✓	sklearn_radius_neighbors_transformer
neural_network	BernoulliRBM	✓	✓	sklearn_bernoulli_rbm
neural_network	MLPClassifier	✓	✓	sklearn_mlp_classifier
neural_network	MLPRegressor	✓	✓	sklearn_mlp_regressor
preprocessing	Binarizer	✓	✓	sklearn_binarizer
preprocessing	FunctionTransformer	✓	✓	sklearn_function_transformer
preprocessing	KBinsDiscretizer	✓	✓	sklearn_kbins_discretizer
preprocessing	KernelCenterer	✓	✓	sklearn_kernel_centerer
preprocessing	LabelBinarizer	✓
preprocessing	LabelEncoder	✓
preprocessing	MaxAbsScaler	✓	✓	sklearn_max_abs_scaler
preprocessing	MinMaxScaler	✓	✓	sklearn_min_max_scaler
preprocessing	MultiLabelBinarizer	✓
preprocessing	Normalizer	✓	✓	sklearn_normalizer
preprocessing	OneHotEncoder	✓	✓	sklearn_one_hot_encoder
preprocessing	OrdinalEncoder	✓	✓	sklearn_ordinal_encoder
preprocessing	PolynomialFeatures	✓	✓	sklearn_polynomial_features
preprocessing	PowerTransformer	✓	✓	sklearn_power_transformer
preprocessing	QuantileTransformer	✓	✓	sklearn_quantile_transformer
preprocessing	RobustScaler	✓	✓	sklearn_robust_scaler
preprocessing	SplineTransformer	✓	✓	sklearn_spline_transformer
preprocessing	StandardScaler	✓	✓	sklearn_standard_scaler
preprocessing	TargetEncoder	✓
semi_supervised	LabelPropagation	✓
semi_supervised	LabelSpreading	✓
semi_supervised	SelfTrainingClassifier	✓
sklearn	AdditiveChi2Sampler	✓	✓	sklearn_additive_chi2_sampler
sklearn	BernoulliNB	✓	✓	sklearn_bernoulli_nb
sklearn	CalibratedClassifierCV	✓	✓	sklearn_calibrated_classifier_cv
sklearn	CategoricalNB	✓	✓	sklearn_categorical_nb
sklearn	ClassifierChain	✓	✓	sklearn_classifier_chain
sklearn	ComplementNB	✓	✓	sklearn_multinomial_nb
sklearn	DummyClassifier	✓	✓	sklearn_dummy_classifier
sklearn	DummyRegressor	✓	✓	sklearn_dummy_regressor
sklearn	FeatureUnion	✓	✓	sklearn_feature_union
sklearn	GaussianNB	✓	✓	sklearn_gaussian_nb
sklearn	GaussianRandomProjection	✓	✓	sklearn_gaussian_random_projection
sklearn	IsotonicRegression	✓	✓	sklearn_isotonic_regression
sklearn	KernelRidge	✓	✓	sklearn_kernel_ridge
sklearn	LinearDiscriminantAnalysis	✓	✓	sklearn_linear_discriminant_analysis
sklearn	MultiOutputClassifier	✓	✓	sklearn_multi_output_classifier
sklearn	MultiOutputRegressor	✓	✓	sklearn_multi_output_regressor
sklearn	MultinomialNB	✓	✓	sklearn_multinomial_nb
sklearn	Nystroem	✓	✓	sklearn_nystroem
sklearn	OneVsOneClassifier	✓	✓	sklearn_one_vs_one_classifier
sklearn	OneVsRestClassifier	✓	✓	sklearn_one_vs_rest_classifier
sklearn	OutputCodeClassifier	✓	✓	sklearn_output_code_classifier
sklearn	Pipeline	✓	✓	sklearn_pipeline
sklearn	PolynomialCountSketch	✓
sklearn	QuadraticDiscriminantAnalysis	✓	✓	sklearn_quadratic_discriminant_analysis
sklearn	RBFSampler	✓	✓	sklearn_rbf_sampler
sklearn	RegressorChain	✓	✓	sklearn_regressor_chain
sklearn	SkewedChi2Sampler	✓
sklearn	SparseRandomProjection	✓
svm	LinearSVC	✓	✓	sklearn_linear_svc
svm	LinearSVR	✓	✓	sklearn_linear_svr
svm	NuSVC	✓	✓	sklearn_svc
svm	NuSVR	✓	✓	sklearn_svr
svm	OneClassSVM	✓	✓	sklearn_one_class_svm
svm	SVC	✓	✓	sklearn_svc
svm	SVR	✓	✓	sklearn_svr
tree	DecisionTreeClassifier	✓	✓	sklearn_decision_tree_classifier
tree	DecisionTreeRegressor	✓	✓	sklearn_decision_tree_regressor
tree	ExtraTreeClassifier	✓	✓	sklearn_extra_tree_classifier
tree	ExtraTreeRegressor	✓	✓	sklearn_extra_tree_regressor

Coverage: 176/197 ~ 89.3%

scikit-survival

<<<

from yobx.sklearn import register_sklearn_converters
from yobx.sklearn.register import get_sklearn_estimator_coverage

register_sklearn_converters()

print(get_sklearn_estimator_coverage(libraries=("sksurv",), rst=True))

>>>

category	estimator	predictable	yobx	converter	since
ensemble	ComponentwiseGradientBoostingSurvivalAnalysis	✓
ensemble	ExtraSurvivalTrees	✓
ensemble	GradientBoostingSurvivalAnalysis	✓
ensemble	RandomSurvivalForest	✓	✓	sklearn_random_survival_forest
linear_model	CoxPHSurvivalAnalysis	✓
linear_model	CoxnetSurvivalAnalysis	✓
linear_model	IPCRidge	✓	✓	sklearn_ipc_ridge
meta	EnsembleSelection	✓
meta	EnsembleSelectionRegressor	✓
meta	MeanEstimator	✓
meta	Stacking	✓
svm	FastKernelSurvivalSVM	✓
svm	FastSurvivalSVM	✓
svm	HingeLossSurvivalSVM	✓
svm	MinlipSurvivalAnalysis	✓
svm	NaiveSurvivalSVM	✓
tree	ExtraSurvivalTree	✓
tree	SurvivalTree	✓

Coverage: 2/18 ~ 11.1%

xgboost

<<<

from yobx.sklearn import register_sklearn_converters
from yobx.sklearn.register import get_sklearn_estimator_coverage

register_sklearn_converters()

print(get_sklearn_estimator_coverage(libraries=("xgboost",), rst=True))

>>>

category	estimator	predictable	yobx	converter	since
xgboost	XGBClassifier	✓	✓	sklearn_xgb_classifier
xgboost	XGBRFClassifier	✓	✓	sklearn_xgb_classifier
xgboost	XGBRFRegressor	✓	✓	sklearn_xgb_regressor
xgboost	XGBRanker	✓	✓	sklearn_xgb_ranker
xgboost	XGBRegressor	✓	✓	sklearn_xgb_regressor

Coverage: 5/5 ~ 100.0%