UNAGI.train.trainer.UNAGI_trainer

class UNAGI.train.trainer.UNAGI_trainer(model, dis_model, modelName, batch_size, epoch_initial, epoch_iter, device, lr, lr_dis, GCN=True, cuda=True)

__init__(model, dis_model, modelName, batch_size, epoch_initial, epoch_iter, device, lr, lr_dis, GCN=True, cuda=True)

Methods

__init__(model, dis_model, modelName, ...[, ...])
get_latent_representation(adata, iteration, ...)	find out the best groups of resolution for clustering
train(adata, iteration, target_dir[, ...])
train_model(adata, vae, dis, train_loader, adj)