def test_rmst_plot_with_single_model(self, block): waltons = load_waltons() kmf = KaplanMeierFitter().fit(waltons["T"], waltons["E"]) rmst_plot(kmf, t=40.0) self.plt.title("test_rmst_plot_with_single_model") self.plt.show(block=block)
def test_rmst_plot_with_two_model(self, block): waltons = load_waltons() ix = waltons["group"] == "control" kmf_con = KaplanMeierFitter().fit(waltons.loc[ix]["T"], waltons.loc[ix]["E"], label="control") kmf_exp = KaplanMeierFitter().fit(waltons.loc[~ix]["T"], waltons.loc[~ix]["E"], label="exp") rmst_plot(kmf_con, model2=kmf_exp, t=40.0) self.plt.title("test_rmst_plot_with_two_model") self.plt.show(block=block)