Commit 16a265ff authored by lucas_miranda's avatar lucas_miranda
Browse files

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

parent 305b1aba
Pipeline #87271 passed with stage
in 21 minutes and 24 seconds
......@@ -426,7 +426,7 @@ for i in range(len(checkpoints) + 1):
else:
encs.append(reducer.fit_transform(predictions[i]))
else:
if red == "LDA":
if red == "LDA" or red == "umap":
encs.append(
reducer.fit_transform(
predictions[i], np.argmax(clusters[i - 1], axis=1)
......
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