Module auton_survival.models.dcm.dcm_torch
Classes
class DeepCoxMixturesTorch (inputdim, k, gamma=1, use_activation=False, layers=None, optimizer='Adam')

PyTorch model definition of the Deep Cox Mixture Survival Model.
The Cox Mixture involves the assumption that the survival function of the individual to be a mixture of K Cox Models. Conditioned on each subgroup Z=k; the PH assumptions are assumed to hold and the baseline hazard rates is determined nonparametrically using an splineinterpolated Breslow's estimator.
Initializes internal Module state, shared by both nn.Module and ScriptModule.
Class variables
var dump_patches : bool
var training : bool
Methods
def forward(self, x) ‑> Callable[..., Any]