Plugin that provides Aliyun oss Artifact Store functionality for MLflow
Project description
Xkool Aliyun OSS store plugin for MLflow
Forked and modified from SeaOfOcean at https://github.com/SeaOfOcean/mlflow-aliyunstore
This repository provides a MLflow plugin that allows users to use Aliyun OSS as the artifact store for MLflow.
Usage
Pip install the package on both your client and the server
pip install mlflow_oss_artifact
Configure environment variables in your OS for Aliyun OSS authentication
Note: checkout this post on stackoverflow to make them permanent if necessary
export MLFLOW_OSS_ENDPOINT_URL=<oss-xx-cityname.aliyuncs.com>
export MLFLOW_OSS_KEY_ID=<your_oss_key_id>
export MLFLOW_OSS_KEY_SECRET=<your_oss_key_secret>
export MLFLOW_OSS_BUCKET_NAME=<your_bucket_name>
To use To use Aliyun OSS as an artifact store, an OSS URI of the form oss://<path>
must be provided, as shown in the example below:
import mlflow
import mlflow.pyfunc
class Mod(mlflow.pyfunc.PythonModel):
def predict(self, ctx, inp):
return 7
exp_name = "myexp"
mlflow.create_experiment(exp_name, artifact_location="oss://mlflow-test/")
mlflow.set_experiment(exp_name)
mlflow.pyfunc.log_model('model_test', python_model=Mod())
In the example provided above, the log_model
operation creates three entries in the OSS storage oss://mlflow-test/$RUN_ID/artifacts/model_test/
, the MLmodel file
and the conda.yaml file associated with the model.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Hashes for mlflow_oss_artifact-2.0.0.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 7576e042feab268047062b2c0ce1b2303add7f91b52f1162f9ba2e6797807c12 |
|
MD5 | 4249703abfb6c7c32ee7bdd7e3f7ff22 |
|
BLAKE2b-256 | e80896efa1dd6da2e2a9b6210ec53ec5df4256094ee306f7e640dfc5ec18b56f |