Module auton_survival.models.dcm.dcm_torch


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 non-parametrically using an spline-interpolated Breslow's estimator.

Initializes internal Module state, shared by both nn.Module and ScriptModule.

Class variables

var dump_patches : bool
var training : bool


def forward(self, x) ‑> Callable[..., Any]