CatBoost Model Type

Upload a CatBoost Model to a Machine Learning Repository

Use the following function to upload a CatBoost model to a Machine Learning repository in Artifactory:

import frogml

repository = "repository-name"
name = "model-name"
namespace = "namespace"
version = "version-1"
properties = {"key1": "value1"}
dependencies = ["path/to/dependencies/pyproject.toml", "path/to/dependencies/poetry.lock"]
code_dir = "full/path/to/code/dir"
catboost_model = get_catboost_model() # Function that Returns a CatBoost model

frogml.catboost.log_model(
    model=catboost_model,
    repository=repository,
    model_name=name,
    namespace=namespace,
    version=version,
    properties=properties,
    dependencies=dependencies,
    code_dir=code_dir,
)

Download a CatBoost Model from a Machine Learning Repository

Use the following function to download a deserialized CatBoost model from a Machine Learning repository in Artifactory:

import frogml

repository = "repository-name"
name = "model-name"
namespace = "namespace"
version = "version-1"
full_target_path = "full/path/to/target/path"

catboost_deserialized_model = frogml.catboost.load_model(
    repository=repository,
    namespace=namespace,
    model_name=name,
    version=version,
    target_path=full_target_path,
)

Get Information on a CatBoost Model in a Machine Learning Repository

Use the frogml.catboost.get_model_info() function to retrieve information on a specific CatBoost model stored in a Machine Learning repository in Artifactory.

frogml.catboost.get_model_info(
    repository=repository,
    name=name,
    namespace=namespace,
    version=version,
)