vak.models.parametric_umap_model.ParametricUMAP

class vak.models.parametric_umap_model.ParametricUMAP(encoder: Module, decoder: Module | None = None, n_neighbors: int = 10, min_dist: float = 0.1, metric: str = 'euclidean', num_epochs: int = 200, lr: float = 0.001, batch_size: int = 64, num_workers: int = 16, random_state: int | None = None)

Bases: object

__init__(encoder: Module, decoder: Module | None = None, n_neighbors: int = 10, min_dist: float = 0.1, metric: str = 'euclidean', num_epochs: int = 200, lr: float = 0.001, batch_size: int = 64, num_workers: int = 16, random_state: int | None = None)

Methods

__init__(encoder[, decoder, n_neighbors, ...])
fit(trainer, dataset_path[, transform])
inverse_transform(Z)
transform(X)