code: def _test_kmeans(
      self,
      n_clusters,
      algorithm="lloyd",
      backend="torch",
      extra_config={},
  ):
      model = KMeans(
          n_clusters=n_clusters, algorithm=algorithm, random_state=0
      )
      np.random.seed(0)
      X = np.random.rand(100, 200)
      X = np.array(X, dtype=np.float32)
  
      model.fit(X)
      torch_model = hummingbird.ml.convert(
          model, backend, X, extra_config=extra_config
      )
      self.assertTrue(torch_model is not None)
      np.testing.assert_allclose(
          model.predict(X), torch_model.predict(X), rtol=1e-6, atol=1e-6
      )
  
imports: import unittest
  import warnings
  import numpy
  import torch
  from sklearn.cluster import KMeans
  import hummingbird.ml
  from hummingbird.ml import constants
  from hummingbird.ml._utils import tvm_installed
label_desc: Use image node_id 1 for calling the TestSklearnClustering obj's underlying member method code with example usage: obj._test_kmeans(n_clusters, algorithm, backend, extra_config) without return types