def survival_y_cox()

in scripts/coxph_preprocessing.py [0:0]

• 8 lines of code
• 3 McCabe index (conditional complexity)

def survival_y_cox(dframe):
    """Returns array of outcome encoded for XGB"""
    y_survival = []

    for idx, row in dframe[["duration", "event"]].iterrows():
        if row["event"]:
            # uncensored
            y_survival.append(int(row["duration"]))
        else:
            # right censored
            y_survival.append(-int(row["duration"]))
    return np.array(y_survival)