vak.nn.loss.umap.umap_loss#

vak.nn.loss.umap.umap_loss(embedding_to: Tensor, embedding_from: Tensor, a, b, negative_sample_rate: int = 5)[source]#

UMAP loss function

Converts distances to probabilities, and then computes cross entropy.