def compute_pred_df()

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


def compute_pred_df(pred_dfs, n_models, n_classes, class_labels):
    pred_cols_for_classes = [PRED_PREFIX + str(c) for c in class_labels] if class_labels is not None else [PRED_COL_IN]
    pred_df = pd.concat([df[pred_cols_for_classes] for df in pred_dfs], axis=1)
    pred_df.columns = compute_pred_col_names(n_models, n_classes)
    return pred_df