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_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
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
