mlsteam-model-sdk 0.4.1

MLSteam Model SDK

mlsteam-model-sdk

SDK for accessing MLSteam models

Setup

pip3 install mlsteam-model-sdk

To process encrypted model versions, install the Themis development package according to the official instrunctions. Debian/Ubuntu users have a handy installation method:

# for users that already have administrator privileges
mlsteam-model-cli install-themisdev

# for those that need privilege lifting
sudo mlsteam-model-cli install-themisdev

Usage

Downloading a model version with SDK

This example assumes the program is running in an MLSteam system. If it is not the case, you may need to setup API_TOKEN with mlsteam-model-cli init --api_token=YOUR_API_TOKEN

You will need administrator privileges to download or load encrypted model versions. For this case, either run the following program with sudo, or enter your password in a sudo prompt during program execution.

sudo python your_program.py

or during execution

[sudo] password for some_username:

Administrator privileges are not required when you only process non-encrypted model versions.

from mlsteam_model_sdk.sdk.model import Model

sdk_model = Model()
sdk_model.download_model_version(project_name='project_owner/project_name',
                                 model_name='model_name',
                                 version_name='version_name')

By default, the model version will be downloaded at $HOME/.mlsteam-model-sdk/models/download/.

This loads a model version and makes prediction:

mv = sdk_model.load_model_version(model_name='model_name',
                                  version_name='version_name')
outputs = mv.predict(inputs)

Importing a model version with CLI

This example assumes the following files are locally available:

1. model version package (required)
2. package encryption key (required only for encrypted packages)

You will need administrator privileges to download or load encrypted model versions, as mentioned in the previous example.

Initialize the SDK settings if you have not done so:

mlsteam-model-cli init

By default, the settings will be at $HOME/.mlsteam-model-sdk/cfg.ini.

To import a package:

# for non-encrypted packages
mlsteam-model-cli mv import-local -f path/to/pkg/file

# for encrypted packages
mlsteam-model-cli mv import-local -f path/to/pkg/file -k path/to/enckey/file

By default, the model and version names to register are read from the package manifest. You may customize these settings with the --model_name and --version_name options.

If the operation is successful, you will find the imported pakage in local model registry:

mlsteam-model-cli mv list-local

   muuid     model_name       vuuid        version_name     puuid     packaged   encrypted      download_time
 ================================================================================================================
 __local__   ...          local-........   ...            __local__   1          ...            .....