Return UMAP graph? about uwot HOT 5 CLOSED

uwot 0.1.8 is on CRAN with this feature available.

from uwot.

Hello, right now the fuzzy simplicial set is not available for output. But it could be added as a new option in the next release.

from uwot.

OK, thanks! I think this would be a great feature for the next release :)

from uwot.

@twhiscock the github version of uwot now has the option to return the fuzzy simplicial set by using:

res <- umap(X, ret_extra = c("fgraph"))

The coordinates will be in res$embedding and the graph in res$fgraph. It's a sparse dgCMatrix from the Matrix package.

This will show up in the next CRAN version, whenever that is.

Note that the graph is further sparsified by dropping low-weight edges that would not be sampled during optimization. That is determined by the n_epochs parameter. If you only care about the graph and not the embedded coordinates, set n_epochs = 0 and no edges are removed.

The effect of the sparsifying is small with default values. I looked at n_epochs = 200 and n_epochs = 500 with MNIST (a largeish dataset) and iris (a very small one) and the number of edges that are dropped is always < 1%.

from uwot.

Awesome :)

from uwot.