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]