def test_qq_plot_left_censoring2(self, block): df = load_lcd() fig, axes = self.plt.subplots(2, 2, figsize=(9, 5)) axes = axes.reshape(4) for i, model in enumerate([WeibullFitter(), LogNormalFitter(), LogLogisticFitter(), ExponentialFitter()]): model.fit_left_censoring(df["T"], df["E"]) ax = qq_plot(model, ax=axes[i]) assert ax is not None self.plt.suptitle("test_qq_plot_left_censoring2") self.plt.show(block=block)
def test_kmf_left_censorship_plots(self, block): kmf = KaplanMeierFitter() lcd_dataset = load_lcd() alluvial_fan = lcd_dataset.loc[lcd_dataset["group"] == "alluvial_fan"] basin_trough = lcd_dataset.loc[lcd_dataset["group"] == "basin_trough"] kmf.fit_left_censoring(alluvial_fan["T"], alluvial_fan["E"], label="alluvial_fan") ax = kmf.plot() kmf.fit_left_censoring(basin_trough["T"], basin_trough["E"], label="basin_trough") ax = kmf.plot(ax=ax) self.plt.title("test_kmf_left_censorship_plots") self.plt.show(block=block) return
def test_kmf_left_censorship_plots(self, block): kmf = KaplanMeierFitter() lcd_dataset = load_lcd() alluvial_fan = lcd_dataset.loc[lcd_dataset['group'] == 'alluvial_fan'] basin_trough = lcd_dataset.loc[lcd_dataset['group'] == 'basin_trough'] kmf.fit(alluvial_fan['T'], alluvial_fan['C'], left_censorship=True, label='alluvial_fan') ax = kmf.plot() kmf.fit(basin_trough['T'], basin_trough['C'], left_censorship=True, label='basin_trough') ax = kmf.plot(ax=ax) self.plt.title("test_kmf_left_censorship_plots") self.plt.show(block=block) return
def test_kmf_left_censorship_plots(self, block): matplotlib = pytest.importorskip("matplotlib") from matplotlib import pyplot as plt kmf = KaplanMeierFitter() lcd_dataset = load_lcd() alluvial_fan = lcd_dataset.ix[lcd_dataset['group'] == 'alluvial_fan'] basin_trough = lcd_dataset.ix[lcd_dataset['group'] == 'basin_trough'] kmf.fit(alluvial_fan['T'], alluvial_fan['C'], left_censorship=True, label='alluvial_fan') ax = kmf.plot() kmf.fit(basin_trough['T'], basin_trough['C'], left_censorship=True, label='basin_trough') ax = kmf.plot(ax=ax) plt.show(block=block) return
def test_kmf_left_censorship_plots(self): matplotlib = pytest.importorskip("matplotlib") from matplotlib import pyplot as plt kmf = KaplanMeierFitter() lcd_dataset = load_lcd() alluvial_fan = lcd_dataset.ix[lcd_dataset['group'] == 'alluvial_fan'] basin_trough = lcd_dataset.ix[lcd_dataset['group'] == 'basin_trough'] kmf.fit(alluvial_fan['T'], alluvial_fan['C'], left_censorship=True, label='alluvial_fan') ax = kmf.plot() kmf.fit(basin_trough['T'], basin_trough['C'], left_censorship=True, label='basin_trough') ax = kmf.plot(ax=ax) plt.show() return