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(max_epochs, ckpt_root, ...[, device])	Returns an instance of lightning.Trainer with a default set of callbacks.
train_parametric_umap_model(model_name, ...)	Train a model from the parametric UMAP family and save results.