vak.datasets.parametric_umap.parametric_umap

A dataset class used to train Parametric UMAP models.

Functions

get_graph_elements(graph, n_epochs)	Get graph elements for Parametric UMAP Dataset.
get_umap_graph(X[, n_neighbors, metric, ...])	Get graph used by UMAP, the fuzzy topological representation.

Classes

ParametricUMAPDataset(dataset_path, ...[, ...])	A dataset class used to train Parametric UMAP models.
ParametricUMAPInferenceDataset(data, dataset_df)