hummingbird.ml.operator_converters.sparkml.vector_assembler¶

Converter for Spark-ML VectorAssembler

hummingbird.ml.operator_converters.sparkml.vector_assembler.convert_sparkml_vector_assembler(operator, device, extra_config)[source]¶

Converter for pyspark.ml.feature.VectorAssembler

Args:: operator: An operator wrapping a pyspark.ml.feature.VectorAssembler 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