ml.mlp_kfold

Module Contents

Classes

MimoMLP

Functions

load_data(mimos, include_esp, data_loc)
validate(model, dataloader, device)
train(feature, layers, lr, n_epochs, l2, ...)
evaluate_model(feature, mlp_cls, test_dataloader, ...)
build_dataloaders(data_split)
run_mlp(data_split_type, include_esp, n_epochs)

ml.mlp_kfold.load_data(mimos, include_esp, data_loc)

ml.mlp_kfold.validate(model, dataloader, device)

ml.mlp_kfold.train(feature, layers, lr, n_epochs, l2, train_dataloader, val_dataloader, device)

ml.mlp_kfold.evaluate_model(feature, mlp_cls, test_dataloader, device, mimos)

class ml.mlp_kfold.MimoMLP(layers)

Bases: torch.nn.Module

forward(x)

ml.mlp_kfold.build_dataloaders(data_split)

ml.mlp_kfold.run_mlp(data_split_type, include_esp, n_epochs)