Module auton_survival.models.cph.dcph_utilities
Functions
def randargmax(b, **kw)-
a random tie-breaking argmax
def partial_ll_loss(lrisks, tb, eb, eps=0.001)def fit_breslow(model, x, t, e)def train_step(model, x, t, e, optimizer, bs=256, seed=100)def test_step(model, x, t, e)def train_dcph(model, train_data, val_data, epochs=50, patience=3, bs=256, lr=0.001, debug=False, random_seed=0, return_losses=False)def predict_survival(model, x, t=None)