vak.train.frame_classificationΒΆ

Function that trains models in the frame classification family.

Functions

get_split_dur(df, split)

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

get_train_callbacks(ckpt_root, ckpt_step, ...)

get_trainer(accelerator, devices, max_steps, ...)

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

train_frame_classification_model(...[, ...])

Train a model from the frame classification family and save results.