def test_kmf_with_interval_censoring_plotting(self, block): kmf = KaplanMeierFitter() left, right = load_diabetes()["left"], load_diabetes()["right"] kmf.fit_interval_censoring(left, right) kmf.plot(color="r") self.plt.show(block=block) return
def test_max_lower_bound_less_than_min_upper_bound(): # https://github.com/CamDavidsonPilon/lifelines/issues/1151 import numpy as np import pandas as pd from lifelines import KaplanMeierFitter # Data np.random.seed(1) left0 = np.random.normal(loc=60, scale=2, size=20) add_time = np.random.normal(loc=100, scale=2, size=10) right1 = left0[0:10] + add_time right0 = right1.tolist() + [np.inf] * 10 # KaplanMeier model = KaplanMeierFitter() model.fit_interval_censoring(lower_bound=left0, upper_bound=right0)