save

Save the quantized pool to a file.

Method call format

save(fname)

Parameters

Parameter Possible types Description Default value
fname string

The name of the output file to save the pool to.

Required parameter
Parameter Possible types Description Default value
fname string

The name of the output file to save the pool to.

Required parameter

Example

Quantize the given dataset and save it to a file:

import numpy as np
from catboost import Pool, CatBoostRegressor


train_data = np.random.randint(1, 100, size=(10000, 10))
train_labels = np.random.randint(2, size=(10000))
quantized_dataset_path = 'quantized_dataset.bin'

# save quantized dataset
train_dataset = Pool(train_data, train_labels)
train_dataset.quantize()
train_dataset.save(quantized_dataset_path)