update umap import

This commit is contained in:
shixian.shi 2024-01-25 10:23:19 +08:00
parent 0281b70896
commit 2d078a26fe

View File

@ -118,7 +118,7 @@ class UmapHdbscan:
self.metric = metric self.metric = metric
def __call__(self, X): def __call__(self, X):
from umap.umap_ import UMAP import umap.umap_ as umap
umap_X = umap.UMAP( umap_X = umap.UMAP(
n_neighbors=self.n_neighbors, n_neighbors=self.n_neighbors,
min_dist=0.0, min_dist=0.0,