hummingbird.ml._utils

Collection of utility functions used throughout Hummingbird.

class hummingbird.ml._utils._Constants(constants, other_constants=None)

Bases: object

Class enabling the proper definition of constants.

hummingbird.ml._utils.assert_lightgbm_installed()

hummingbird.ml._utils.assert_onnx_ml_tools_installed()

hummingbird.ml._utils.assert_onnx_runtime_installed()

hummingbird.ml._utils.assert_pandas_installed()

hummingbird.ml._utils.assert_sklearn_installed()

hummingbird.ml._utils.assert_sparkml_installed()

hummingbird.ml._utils.assert_torch_installed()

hummingbird.ml._utils.assert_tvm_installed()

hummingbird.ml._utils.assert_xgboost_installed()

hummingbird.ml._utils.check_dumped_versions(configurations, *args)

When a model is loaded this function is used to check that the versions of the modules used at saving time match with the version at loading time.

hummingbird.ml._utils.dump_versions(*args)

Utility function used to generate a string containing the versions of the main modules used to convert a model.

hummingbird.ml._utils.from_strings_to_ints(input, max_string_length)

Utility function used to transform string inputs into a numerical representation.

hummingbird.ml._utils.get_device(model)

Convenient function used to get the runtime device for the model.

hummingbird.ml._utils.is_pandas_dataframe(df)

hummingbird.ml._utils.is_spark_dataframe(df)

hummingbird.ml._utils.lightgbm_installed()

Checks that LightGBM is available.

hummingbird.ml._utils.load(location, digest=None)

Utility function used to load arbitrary Hummingbird models.

Args:

location: The location of the model. digest (optional): A digest string to verify the model integrity (created during save).

hummingbird.ml._utils.onnx_ml_tools_installed()

Checks that ONNXMLTools is available.

hummingbird.ml._utils.onnx_runtime_installed()

Checks that ONNX Runtime is available.

hummingbird.ml._utils.pandas_installed()

Checks that Pandas is available.

hummingbird.ml._utils.prophet_installed()

Checks that Prophet is available.

hummingbird.ml._utils.sklearn_installed()

Checks that Sklearn is available.

hummingbird.ml._utils.sparkml_installed()

Checks that Spark ML/PySpark is available.

hummingbird.ml._utils.torch_installed()

Checks that PyTorch is available.

hummingbird.ml._utils.tvm_installed()

Checks that TVM is available.

hummingbird.ml._utils.xgboost_installed()

Checks that XGBoost is available.