← retour aux snippets

catboost: sauvegarder/charger modèle

Persister un modèle CatBoost au format binaire .cbm.

objectif

Persister un modèle CatBoost au format binaire .cbm.

code minimal

from catboost import CatBoostClassifier

model = CatBoostClassifier(iterations=100, depth=6, learning_rate=0.1, loss_function="Logloss", random_seed=0, verbose=False)
X = [[0,1],[1,0],[1,1],[0,0]]; y = [0,1,1,0]
model.fit(X, y)
model.save_model("model.cbm")
loaded = CatBoostClassifier()
loaded.load_model("model.cbm")
print(hasattr(loaded, "predict_proba"))

utilisation

# Export JSON
model.save_model("model.json", format="json")
print(True)

variante(s) utile(s)

# Convertir pour CoreML/ONNX (via outils externes CatBoost)
print(True)

notes

  • Le format .cbm est le plus complet; versionnez vos modèles et paramètres associés.