vak.train.parametric_umapΒΆ

Function that trains models in the Parametric UMAP family.

Functions

get_split_dur(df, split)

Get duration of a split in a dataset from a pandas DataFrame representing the dataset.

get_trainer(accelerator, devices, ...)

Returns an instance of lightning.pytorch.Trainer with a default set of callbacks.

train_parametric_umap_model(model_config, ...)

Train a model from the parametric UMAP family and save results.