Merge pull request #101 from sparks-baird/umap-hdbscan-kwargs

cluster when it was supposed to be vis
sparks-baird · Aug 13, 2022 · c0347e7 · c0347e7
2 parents f1c80c4 + 1357938
commit c0347e7
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/mat_discover/mat_discover_.py b/mat_discover/mat_discover_.py
@@ -1230,10 +1230,10 @@ def umap_fit_vis(self, dm, random_state=None):
         """
         if random_state is not None:
             self.umap_vis_kwargs["random_state"] = random_state
-        self.umap_cluster_kwargs["densmap"] = True
-        self.umap_cluster_kwargs["output_dens"] = True
-        self.umap_cluster_kwargs["dens_lambda"] = self.dens_lambda
-        self.umap_cluster_kwargs["low_memory"] = False
+        self.umap_vis_kwargs["densmap"] = True
+        self.umap_vis_kwargs["output_dens"] = True
+        self.umap_vis_kwargs["dens_lambda"] = self.dens_lambda
+        self.umap_vis_kwargs["low_memory"] = False
         with self.Timer("fit-vis-UMAP"):
             std_trans = umap.UMAP(**self.umap_vis_kwargs).fit(dm)
         return std_trans