load_model

Load the model from a file.

Method call format

load_model(fname, format='cbm')

Parameters

fname

Description

The path to the input model.

Possible types

string

Default value

Required parameter

format

Description

The format of the input model.

Possible values:

  • cbm — CatBoost binary format.
  • AppleCoreML(only datasets without categorical features are currently supported).
  • json — JSON format. Refer to the CatBoost JSON model tutorial for format details.
  • onnx — ONNX-ML format (only datasets without categorical features are currently supported). Refer to https://onnx.ai/ for details. See the ONNX section for details on applying the resulting model.
  • CpuSnapshot — CatBoost training snapshot format (only CPU and datasets without categorical features are currently supported).

Possible types

string

Default value

cbm

Usage examples

from catboost import CatBoostClassifier, Pool

train_data = [[1, 3],
              [0, 4],
              [1, 7]]
train_labels = [1, 0, 1]

# catboost_pool = Pool(train_data, train_labels)

model = CatBoostClassifier(learning_rate=0.03)
model.fit(train_data,
          train_labels,
          verbose=False)

model.save_model("model")

from_file = CatBoostClassifier()

from_file.load_model("model")