Log scikit-learn Models

Run the following command:

import frogml

repository = "<REPO_NAME>"
name = "<MODEL_NAME>"
version = "<MODEL_VERSION>" # optional
properties = {"<KEY1>": "<VALUE1>"} # optional
dependencies = ["<MODEL_DEPENDENCIES>"] # optional
code_dir = "<CODE_PATH>" # optional
parameters = {"<HYPERPARAM_NAME>": "<VALUE>"} # optional
metrics = {"<METRIC_NAME>": "<VALUE>"} # optional
predict_file = "<PREDICT_PATH>" # optional
joblib_model = get_scikit_learn()

frogml.scikit_learn.log_model(
    model=joblib_model,
    repository=repository,
    model_name=name,
    version=version,
    properties=properties,
    dependencies=dependencies,
    code_dir=code_dir,
    parameters=parameters,
    metrics=metrics,
    predict_file=predict_file,
)

Important

The parameters code_dir, dependencies, and predict_file must be provided as a complete set. You must specify all of them or none at all.

Where:

<REPO_NAME>: The name of the Artifactory repository where the model is stored
<MODEL_NAME>: The unique name of the model you want to download
<MODEL_VERSION> (Optional): The version of the model you want to upload. If version is not specified, the current timestamp is used
<KEY1> and <VALUE1> (Optional): Properties key pair values to add searchable string-based tags or metadata to the model. Separate multiple key pairs with a comma
<MODEL_DEPENDENCIES> (Optional): Dependencies required to run the model, in one of the following formats:
- A list of specific package requirements, for example ["catboost==1.2.5", "scikit-learn==1.3.2"]
- A path to a single requirements.txt , pyproject.toml, or conda.yml package manager file
- A two-item list containing paths to the pyproject.toml and poetry.lock files
<CODE_PATH> (Optional): The path to the directory containing the source code
<HYPERPARAM_NAME> and <VALUE> (Optional): Hyperparameters used to train the model
<METRIC_NAME> and <VALUE> (Optional): Metrics key pair values to add searchable numeric metadata to the model. Separate multiple key pairs with a comma
<PREDICT_PATH> (Optional): The path to a script that defines how to make predictions with the model.

For example:

import frogml

repository = "ml-local"
name = "customer-churn-predictor"
version = "1.2.0"
properties = {"feature_set_version": "v3"}
dependencies = ["scikit-learn==1.4.2", "pandas==2.2.2"]
code_dir = "src/training/"
predict_file = "src/training/predict.py"
joblib_model = get_scikit_learn()

frogml.scikit_learn.log_model(
    model=joblib_model,
    repository=repository,
    model_name=name,
    version=version,
    properties=properties,
    dependencies=dependencies,
    code_dir=code_dir,
)