Commit e99ef7a8 authored by lucas_miranda's avatar lucas_miranda
Browse files

Refactored a bit the old visualization pipeline - added umap to requirements

parent 079198bb
Pipeline #87251 failed with stage
in 21 minutes and 42 seconds
......@@ -405,7 +405,7 @@ print("Reducing latent space to 2 dimensions for dataviz...")
if red == "LDA":
reducer = LinearDiscriminantAnalysis(n_components=2)
elif red == "UMAP":
elif red == "umap":
reducer = umap.UMAP(n_components=2)
elif red == "tSNE":
......@@ -420,7 +420,7 @@ for i in range(len(checkpoints) + 1):
if variational
else np.zeros(samples)
)
if red == "LDA":
if red == "LDA" or red == "umap":
encs.append(reducer.fit_transform(predictions[i], clusts))
else:
encs.append(reducer.fit_transform(predictions[i]))
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment