26b41f186c9288785e2045bd1cb42fb4503a7bb7,examples/cure_model.py,,,#,24

Before Change

swf = CureModel(penalizer=0.0)

rossi = load_rossi()
rossi["intercept"] = 1.0

covariates = {"lambda_": rossi.columns, "rho_": ["intercept"], "beta_": rossi.columns}

swf.fit(rossi, "week", event_col="arrest", regressors=covariates)  // TODO: name

After Change

covariates = {
    "lambda_": rossi.columns.difference(["week", "arrest"]),
    "rho_": ["1"],
    "beta_": rossi.columns.difference(["week", "arrest"]),
}

swf.fit(rossi, "week", event_col="arrest", regressors=covariates)  // TODO: name

Instances