save

Save the quantized pool to a file.

Method call format

save(fname)

Parameters

fname

Description

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

Possible types

string

Default value

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)