def test_survival_probability_calibration_on_out_of_sample_data(self, block): rossi = load_rossi() rossi = rossi.sample(frac=1.0) cph = CoxPHFitter().fit(rossi.loc[:300], "week", "arrest") survival_probability_calibration(cph, rossi.loc[300:], 25) self.plt.title("test_survival_probability_calibration_on_out_of_sample_data") self.plt.show(block=block)
def test_survival_probability_calibration(self, block): rossi = load_rossi() cph = CoxPHFitter().fit(rossi, "week", "arrest") survival_probability_calibration(cph, rossi, 25) self.plt.title("test_survival_probability_calibration") self.plt.show(block=block)