如何将h2o4gpu Kmeans对象转换为？

Question

我所处的位置需要转换，然后将h2o4gpu保存为只是一个sklearn对象。

我想也许我可以这么做？我原以为我能够保存sklearn_model并加载它，但我得到了错误: AttributeError：'KMeans‘对象没有属性'_n_threads’

from h2o4gpu.solvers import KMeans as GPUKMeans
from sklearn.cluster import KMeans

...

gpu_model = GPUKMeans(n_clusters=num_clusters)

gpu_model.fit(embeddings)

sklearn_model = KMeans(n_clusters=num_clusters)
sklearn_model.cluster_centers_= gpu_model.cluster_centers_;

...

Answer 1

在深入研究源代码之后，我发现一些代码正在做类似的事情：

from h2o4gpu.solvers import KMeans as GPUKMeans
from sklearn.cluster import KMeans
from sklearn.utils._openmp_helpers import _openmp_effective_n_threads    

...

gpu_model = GPUKMeans(n_clusters=num_clusters)

gpu_model.fit(embeddings)

kmeans_model = KMeans(n_clusters=num_clusters)
kmeans_model.cluster_centers_= gpu_model.cluster_centers_;
kmeans_model.labels_= gpu_model.labels_;
kmeans_model.inertia_= gpu_model.inertia_;
kmeans_model._n_threads = _openmp_effective_n_threads()



...