hummingbird.ml.operator_converters.sklearn.cluster¶

Converters for scikit-learn KMeans and MeanShift models.

class hummingbird.ml.operator_converters.sklearn.cluster.KMeans(*args: Any, **kwargs: Any)[source]¶

Bases: PhysicalOperator, Module

Class implementing Kmeans in PyTorch

_abc_impl = <_abc_data object>¶

forward(x)[source]¶

hummingbird.ml.operator_converters.sklearn.cluster.convert_sklearn_kmeans_model(operator, device, extra_config)[source]¶