def test_weibull_aft_plot_covariate_groups(self, block): df = load_rossi() aft = WeibullAFTFitter() aft.fit(df, "week", "arrest") aft.plot_covariate_groups("age", [10, 50, 80]) self.plt.tight_layout() self.plt.title("test_weibull_aft_plot_covariate_groups") self.plt.show(block=block)
def test_weibull_aft_plot_covariate_groups_with_multiple_columns(self, block): df = load_rossi() aft = WeibullAFTFitter() aft.fit(df, "week", "arrest") aft.plot_covariate_groups(["age", "prio"], [[10, 0], [50, 10], [80, 50]]) self.plt.tight_layout() self.plt.title("test_weibull_aft_plot_covariate_groups_with_multiple_columns") self.plt.show(block=block)