ignore npy

.gitignore

@@ -7,3 +7,4 @@ out/
 .pat
 out*
 .lr*
+*.npy

makefile

@@ -55,6 +55,8 @@ umap:
 		python scripts/sortcolor.py -s umap --dpi 300 --seed $$seed ; \
 	done
 
+sort_umap:;
+	python scripts/sortcolor.py -s umap --dpi 300 --seed 21
 
 sort_lex:
 	python scripts/sortcolor.py -s lex --dpi 300

scripts/requirements.cuml.sh

@@ -0,0 +1,3 @@
+pip install \
+    --extra-index-url=https://pypi.nvidia.com \
+    cudf-cu12==23.12.* cuml-cu12==23.12.*

scripts/requirements.rapids.sh

@@ -0,0 +1,6 @@
+pip install \
+    --extra-index-url=https://pypi.nvidia.com \
+    cudf-cu12==23.12.* dask-cudf-cu12==23.12.* cuml-cu12==23.12.* \
+    cugraph-cu12==23.12.* cuspatial-cu12==23.12.* cuproj-cu12==23.12.* \
+    cuxfilter-cu12==23.12.* cucim-cu12==23.12.* pylibraft-cu12==23.12.* \
+    raft-dask-cu12==23.12.*

scripts/sortcolor.py

@@ -79,15 +79,16 @@ if KIND in ("lex", "alpha", "abc"):
     preds = np.array(colors)
 
 elif KIND == "umap":
-    import umap
+    # from umap import UMAP
+    from cuml import UMAP
 
     # Use UMAP to create a 1D representation
-    reducer = umap.UMAP(
+    reducer = UMAP(
         n_components=1,
         n_neighbors=250,
         min_dist=0.005,
         metric="euclidean",
-        # random_state=SEED,
+        random_state=SEED,
     )
     embedding = reducer.fit_transform(np.array(rgb_values))