aeaeb8c961e99d60b60a65e519e4b0c61747942c,perf_tests/cp_perf_test.py,,,#,5

Before Change

    df = pd.concat([df] * reps)
    print(df.shape)

    cph = CoxPHFitter(baseline_estimation_method="spline", n_baseline_knots=1)
    start_time = time.time()
    cph.fit(df, duration_col="week", event_col="arrest", show_progress=True, strata=["mar", "paro"])
    cph.print_summary()
    r = cph.predict_cumulative_hazard(df)

After Change

    df = pd.concat([df] * reps)
    print(df.shape)

    cph = CoxPHFitter(strata=["wexp"])
    start_time = time.time()
    cph.fit(df, duration_col="week", event_col="arrest", show_progress=True)
    cph.print_summary()

Instances