hummingbird.ml.operator_converters._array_feature_extractor_implementations

Base class for array feature extractor

class hummingbird.ml.operator_converters._array_feature_extractor_implementations.ArrayFeatureExtractor(*args: Any, **kwargs: Any)[source]

Bases: PhysicalOperator, Module

Class implementing ArrayFeatureExtractor in PyTorch

This is used by SelectKBest, VarianceThreshold operators in scikit-learn

_abc_impl = <_abc_data object>
forward(x)[source]