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

Before Change

    reps = 1
    df = load_rossi()
    df = pd.concat([df] * reps)
    cp_breslow = CoxPHFitter(penalizer=0.1, l1_ratio=1.0, baseline_estimation_method="spline")
    start_time = time.time()
    cp_breslow.fit(df, duration_col="week", event_col="arrest", show_progress=True)
    print("--- %s seconds ---" % (time.time() - start_time))
    cp_breslow.print_summary(2)

After Change

    reps = 1
    df = load_rossi()
    df = pd.concat([df] * reps)
    cph = CoxPHFitter()
    start_time = time.time()
    cph.fit(df, duration_col="week", event_col="arrest", show_progress=True)
    print("--- %s seconds ---" % (time.time() - start_time))
    cph.print_summary(2)
    print(cph.compute_followup_hazard_ratios(df, [15, 20, 30, 40, 50, 52]))
    print(cph.hazard_ratios_)
    cph.compute_followup_hazard_ratios(df, [15, 20, 30, 40, 50, 52]).plot()

Instances