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)[source]¶
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)[source]¶
Methods
__init__
(encoder[, decoder, n_neighbors, ...])fit
(trainer, dataset_path[, transform])inverse_transform
(Z)transform
(X)