hummingbird.ml.containers._input_containers¶

Input containers used to wrap input models are listed here. These containers are used to represent different input models (e.g., onnx, spark-ml) within Hummingbird in a uniform way.

class hummingbird.ml.containers._input_containers.CommonONNXModelContainer(onnx_model)[source]¶

Bases: CommonSklearnModelContainer

Common container for input ONNX operators.

class hummingbird.ml.containers._input_containers.CommonSparkMLModelContainer(sparkml_model)[source]¶

Bases: CommonSklearnModelContainer

Common container for input Spark-ML operators.