hummingbird.ml.operator_converters.sparkml.linear¶
Converters for SparkML linear models: LogisticRegressionModel
- hummingbird.ml.operator_converters.sparkml.linear.convert_sparkml_linear_model(operator, device, extra_config)[source]¶
Converter for pyspark.ml.classification.LogisticRegressionModel
- Args:
operator: An operator wrapping a pyspark.ml.classification.LogisticRegressionModel 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