def compute_models_meta_data()

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


def compute_models_meta_data(pred_dfs):
    n_models = len(pred_dfs)

    pred_cols_non_class = [PRED_INDEX, TARGET_INDEX, PRED_COL_IN, TARGET_COL_IN]
    classes = [c.replace(PRED_PREFIX, '')
               for c in pred_dfs[0].columns if c.startswith(PRED_PREFIX) and
               not c in pred_cols_non_class and not c.startswith(INDEXED_FEATURE_PREFIX)]

    n_classes = len(classes) if len(classes) != 0 else 1
    class_labels = classes if len(classes) != 0 else None

    return n_models, n_classes, class_labels