INNER CODE UNIT · Python

predictor

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

        predictor = init_model._to_predictor(dict(init_model.params, **params))
    else:
        predictor = None
    init_iteration = predictor.num_total_iteration if predictor is not None else 0
    # check dataset
    if not isinstance(train_set, Dataset):
        raise TypeError("Training only accepts Dataset object")

    use_ar1_mf_fidelity_mean = gp_model is not None and gp_model.fidelity_specific_mean

    if use_ar1_mf_fidelity_mean:
        if gp_model.gp_coords is None:
            raise ValueError("Fidelity-specific GPBoost means require retained GP coordinates")
        _add_ar1_mf_fidelity_feature(train_set, gp_model.gp_coords[:, -1])
        if feature_name != 'auto':
            feature_name = list(feature_name) + ["AR1_MF_fidelity"]

    train_set._update_params(params) \

View source record →