hummingbird.ml.operator_converters.sklearn.kneighbors

Converters for scikit-learn k neighbor models: KNeighborsClassifier, KNeighborsRegressor.

hummingbird.ml.operator_converters.sklearn.kneighbors._convert_kneighbors_model(operator, device, extra_config, is_classifier)

hummingbird.ml.operator_converters.sklearn.kneighbors.convert_sklearn_kneighbors_classification_model(operator, device, extra_config)

Converter for sklearn.neighbors.KNeighborsClassifier

Args:

operator: An operator wrapping a sklearn.neighbors.KNeighborsClassifier 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

hummingbird.ml.operator_converters.sklearn.kneighbors.convert_sklearn_kneighbors_regression_model(operator, device, extra_config)

Converter for sklearn.neighbors.KNeighborsRegressor

Args:

operator: An operator wrapping a sklearn.neighbors.KNeighborsRegressor 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