hummingbird.ml.operator_converters.sparkml.discretizer¶
Converter for Spark-ML discretizers: Bucketizer.
- hummingbird.ml.operator_converters.sparkml.discretizer.convert_sparkml_bucketizer(operator, device, extra_config)[source]¶
Converter for pyspark.ml.feature.Bucketizer
- Args:
operator: An operator wrapping a pyspark.ml.feature.QuantileDiscretizer 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