Trainer

coral.trainer.train.train_model(model, optimizer, dataloader, epochs=300, device='cpu', prot_idx=None, gene_idx=None)

class coral.trainer.train.NegBinom(*args: Any, **kwargs: Any)

Gamma-Poisson mixture approximation of Negative Binomial(mean, dispersion)

lambda ~ Gamma(mu, theta) x ~ Poisson(lambda)