Function make_reference_data

catboost/python-package/ut/medium/test.py:4345–4350

(pool, calc_shap_mode)

def make_reference_data(pool, calc_shap_mode):
    reference_data = None
    if calc_shap_mode == "IndependentTreeSHAP":
        quarter_size = pool.num_row() // 4
        reference_data = pool.slice(list(range(quarter_size)))
    return reference_data


@pytest.mark.parametrize('calc_shap_mode', ['TreeSHAP', 'IndependentTreeSHAP'])