Update

mlexchange · Nov 13, 2023 · 35567cd · 35567cd
1 parent d320236
commit 35567cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -14,6 +14,8 @@ First, build the dimension reduction image in terminal:
 
 Once built, you can run the following examples:
 `make UMAP_example`
+which is equivalend to first `make run_docker` then `python umap_run.py data/example_shapes/Demoshapes.npz data/output '{"n_components": 2, "min_dist": 0.1, "n_neighbors": 7}'`.
+
 
 These examples utilize the information stored in the folder /data. The computed latent vectors will be saved in data/output.
 

diff --git a/umap_run.py b/umap_run.py
@@ -44,7 +44,7 @@ def computeUMAP(data,
     images = None
     if images_dir == "data/example_shapes/Demoshapes.npz":
         images = np.load(images_dir)['arr_0']
-    if images_dir == "data/example_latentrepresentation/f_vectors.parquet":
+    elif images_dir == "data/example_latentrepresentation/f_vectors.parquet":
         df = pd.read_parquet(images_dir)
         images = df.values
     else: # user uploaded zip file