Module auton_survival.models.dsm.utilities
Utility functions to train the Deep Survival Machines models
Functions
def get_optimizer(model, lr)def pretrain_dsm(model, t_train, e_train, t_valid, e_valid, n_iter=10000, lr=0.01, thres=0.0001)def train_dsm(model, x_train, t_train, e_train, x_valid, t_valid, e_valid, n_iter=10000, lr=0.001, elbo=True, bs=100, random_seed=0)-
Function to train the torch instance of the model.