SpatialFuser.train

Functions

train_emb(args, dataLoader)

Train the MCGATE model.

train_integration(args_list, dataLoader_list)

Train the entire SpatialFuser framework, including slice-specific MCGATEs, the matching layer, and the fusion layer.

Classes

emb_trainer(args, loader, adata)

Trainer of the MCGATE model.

integration_trainer(args_list, dataLoader_list)

Trainer of the entire SpatialFuser framework, including slice-specific MCGATEs, the matching layer, and the fusion layer.