Types

Type definitions for the jmstate package.

BucketData(idxs, t0, t1)	NamedTuple representing a set of transitions for visualization purposes.
IndividualParametersFn(*args, **kwargs)	Protocol defining the individual parameters function.
LinkFn(*args, **kwargs)	Protocol defining a link function for multistate models.
LogBaseHazardFn(*args, **kwargs)	Abstract base class for log base hazard functions.
ModelData(x, t, y, trajectories, c)	Dataclass containing learnable multistate joint model data.
ModelDesign(indiv_params_fn, regression_fn, ...)	Dataclass encapsulating the design of a multistate joint model.
ModelParameters(fixed_effects, random_prec, ...)	nn.Module encapsulating all model parameters for a multistate joint model.
PrecisionParameters(flat, dim, precision_type)	nn.Module encapsulating precision matrix parameters.
RegressionFn(*args, **kwargs)	Protocol defining a regression function for multistate models.
SampleData(x, trajectories, indiv_params[, ...])	Dataclass containing individual-level data for sampling procedures.