def test_predict_cumulative_hazard_inputs(self, data_pred1): aaf = AalenAdditiveFitter() aaf.fit(data_pred1, duration_col='t', event_col='E',) x = data_pred1.ix[:5].drop(['t', 'E'], axis=1) y_df = aaf.predict_cumulative_hazard(x) y_np = aaf.predict_cumulative_hazard(x.values) assert_frame_equal(y_df, y_np)
def test_nn_cumulative_hazard_will_set_cum_hazards_to_0(self, rossi): aaf = AalenAdditiveFitter(nn_cumulative_hazard=False) aaf.fit(rossi, event_col='arrest', duration_col='week') cum_hazards = aaf.predict_cumulative_hazard(rossi) assert (cum_hazards < 0).stack().mean() > 0 aaf = AalenAdditiveFitter(nn_cumulative_hazard=True) aaf.fit(rossi, event_col='arrest', duration_col='week') cum_hazards = aaf.predict_cumulative_hazard(rossi) assert (cum_hazards < 0).stack().mean() == 0