vak.nn.loss.umap#

Parametric UMAP loss function.

Functions

compute_cross_entropy(probabilities_graph, ...)

Computes cross entropy as used for UMAP cost function

convert_distance_to_probability(distances[, ...])

Convert distances to probability.

umap_loss(embedding_to, embedding_from, a, b)

UMAP loss function

Classes

UmapLoss([spread, min_dist, ...])