INNER CODE UNIT · Python

_agg_cv_result

fabsig/GPBoost · python-package/gpboost/engine.py:568

def _agg_cv_result(raw_results, eval_train_metric=False):
    """Aggregate cross-validation results."""
    cvmap = collections.OrderedDict()
    metric_type = {}
    for one_result in raw_results:
        for one_line in one_result:
            if eval_train_metric:
                key = "{} {}".format(one_line[0], one_line[1])
            else:
                key = one_line[1]
            metric_type[key] = one_line[3]
            cvmap.setdefault(key, [])
            cvmap[key].append(one_line[2])
    return [('cv_agg', k, np.mean(v), metric_type[k], np.std(v)) for k, v in cvmap.items()]


def cv(params, train_set, gp_model=None, num_boost_round=1000, early_stopping_rounds=None,
       folds=None, nfold=5, metric=None, verbose_eval=None,

View source record →