Function check_save_quantized_pool

catboost/python-package/ut/medium/test.py:8795–8818

(train_pool, train_pool_copy, test_pool, params)

def check_save_quantized_pool(train_pool, train_pool_copy, test_pool, params):
    feature_names = train_pool.get_feature_names()

    train_quantized_pool = train_pool_copy
    train_quantized_pool.quantize()

    assert (train_quantized_pool.is_quantized())

    train_quantized_pool.save(OUTPUT_QUANTIZED_POOL_PATH)

    train_quantized_load_pool = Pool(get_quantized_path(OUTPUT_QUANTIZED_POOL_PATH))

    model = CatBoost(params=params)
    model_fitted_with_load_quantized_pool = CatBoost(params=params)

    model.fit(train_pool)
    predictions1 = model.predict(test_pool)

    model_fitted_with_load_quantized_pool.fit(train_quantized_load_pool)
    predictions2 = model_fitted_with_load_quantized_pool.predict(test_pool)

    loaded_feature_names = train_quantized_load_pool.get_feature_names()
    assert (feature_names == loaded_feature_names)
    assert all(predictions1 == predictions2)


FEATURES_TYPES = ['num', 'cat', 'num_and_cat']