コード例 #1
0
 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)
コード例 #2
0
ファイル: test_estimation.py プロジェクト: nerdless/lifelines
 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)
コード例 #3
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
コード例 #4
0
ファイル: test_estimation.py プロジェクト: DGaffney/lifelines
    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