to_regressor
Purpose
Convert a model of type CatBoost to a model of type CatBoostRegressor. The model can be converted if the loss function of the source model is compatible with the one of the resulting model.
Method call format
to_regressor(model)
Parameters
model
Description
The input CatBoost model for convert.
Possible types
catboost.core.CatBoost
Default value
Obligatory parameter
Type of return value
catboost.core.CatBoostRegressor
Example
from catboost import Pool, to_regressor, CatBoost
train_data = [[0, 3],
[4, 1],
[8, 1],
[9, 1]]
train_labels = [0, 0, 1, 1]
model = CatBoost(params={'loss_function': 'Logloss'})
model.fit(train_data,
train_labels,
verbose=False)
print("Source model type: ", type(model))
converted_model = to_regressor(model)
print("Converted model type: ", type(converted_model))
The output of this example:
('Source model type: ', <class 'catboost.core.CatBoost'>)
('Converted model type: ', <class 'catboost.core.CatBoostRegressor'>)