Skip to content

Commit 32f80a1

Browse files
committed
resolve comments
1 parent b79f0a1 commit 32f80a1

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

docs/source/user_guide/model_registration/quick_start.rst

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -321,7 +321,10 @@ Other Frameworks
321321
preds = catboost_estimator.predict(X_test)
322322
323323
# Instantiate ads.model.generic_model.GenericModel using the trained Custom Model using the trained CatBoost Classifier model
324-
catboost_model = GenericModel(estimator=catboost_estimator, artifact_dir=tempfile.mkdtemp())
324+
catboost_model = GenericModel(estimator=catboost_estimator,
325+
artifact_dir=tempfile.mkdtemp(),
326+
model_save_serializer="cloudpickle",
327+
model_input_serializer="json")
325328
326329
# Autogenerate score.py, pickled model, runtime.yaml, input_schema.json and output_schema.json
327330
catboost_model.prepare(

0 commit comments

Comments
 (0)