hummingbird.ml.operator_converters._kneighbors_implementations

Base classes for KNeighbors model implementations: (KNeighborsClassifier, KNeighborsRegressor).

class hummingbird.ml.operator_converters._kneighbors_implementations.KNeighborsModel(*args: Any, **kwargs: Any)[source]

Bases: PhysicalOperator, Module

_abc_impl = <_abc_data object>
forward(x)[source]
class hummingbird.ml.operator_converters._kneighbors_implementations.MetricType(value)[source]

Bases: Enum

An enumeration.

mahalanobis = 4[source]
minkowski = 1[source]
seuclidean = 3[source]
wminkowski = 2[source]