pylissom.utils.training

Submodules

pylissom.utils.training.cross_validation module

class pylissom.utils.training.cross_validation.CVSubjectIndependent(ck_dataset, k=5)[source]

Bases: object

train_val_samplers()[source]
test_sampler()[source]
pylissom.utils.training.cross_validation.run_cross_validation(model_fn, ck_dataset, cv_sampler, args)[source]

pylissom.utils.training.pipeline module

class pylissom.utils.training.pipeline.Pipeline(model, optimizer=None, loss_fn=None, log_interval=10, dataset_len=None, cuda=False, prefix='', use_writer=False)[source]

Bases: object

train(train_data_loader, epoch)[source]
test(test_data_loader, epoch)[source]
static process_input(inp, normalize=False)[source]
accuracy(data_loader)[source]