def compute_target_df()

in bindings/python-compute/data_manager.py [0:0]

• 12 lines of code
• 4 McCabe index (conditional complexity)

def compute_target_df(pred_dfs, n_models, use_same_target=True):
    if use_same_target:
        target_col = pred_dfs[0][TARGET_COL_IN]
        target_col.name = TARGET_COL_OUT(0)
        return pd.DataFrame(target_col)

    target_cols = []
    for m in range(n_models):
        if TARGET_COL_IN in pred_dfs[m].columns:
            target_col = pred_dfs[m][TARGET_COL_IN]
            target_col.name = TARGET_COL_OUT(m)
            target_cols.append(target_col)
    return pd.concat(target_cols, axis=1)