hummingbird.ml.operator_converters.sklearn.array_feature_extractor

Converters for scikit-learn feature selectors: SelectKBest, SelectPercentile, VarianceThreshold.

hummingbird.ml.operator_converters.sklearn.array_feature_extractor.convert_sklearn_select_k_best(operator, device, extra_config)

Converter for sklearn.feature_selection.SelectKBest.

Args:

operator: An operator wrapping a sklearn.feature_selection.SelectKBest model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy

Returns:

A PyTorch model

hummingbird.ml.operator_converters.sklearn.array_feature_extractor.convert_sklearn_select_percentile(operator, device, extra_config)

Converter for sklearn.feature_selection.SelectPercentile.

Args:

operator: An operator wrapping a sklearn.feature_selection.SelectPercentile model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy

Returns:

A PyTorch model

hummingbird.ml.operator_converters.sklearn.array_feature_extractor.convert_sklearn_variance_threshold(operator, device, extra_config)

Converter for sklearn.feature_selection.VarianceThreshold.

Args:

operator: An operator wrapping a sklearn.feature_selection.VarianceThreshold model device: String defining the type of device the converted operator should be run on extra_config: Extra configuration used to select the best conversion strategy

Returns:

A PyTorch model