A MLflow plugin that allows users to use Baidu BCE BOS as the artifact store for MLflow.
Project description
mlflow-baidubce-store-plugin
A MLflow plugin that allows users to use Baidu BCE BOS(CN/EN) as the artifact store for MLflow.
Example
pip install mlflow-baidubce-store-plugin -U
import mlflow.pyfunc
import os
os.environ["MLFLOW_BOS_ENDPOINT"] = "bj.bcebos.com"
os.environ["MLFLOW_BOS_SECRET_ACCESS_KEY"] = "AK"
os.environ["MLFLOW_BOS_KEY_ID"] = "SK"
class Mod(mlflow.pyfunc.PythonModel):
def predict(self, ctx, inp):
return 8765
exp_name = "bos-exp"
mlflow.create_experiment(exp_name, artifact_location="bos://mlflow-test/")
mlflow.set_experiment(exp_name)
mlflow.pyfunc.log_model('model_test', python_model=Mod())
print(mlflow.get_artifact_uri())
mlflow.artifacts.download_artifacts(mlflow.get_artifact_uri())
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
Built Distribution
Close
Hashes for mlflow-baidubce-store-plugin-1.0.7.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | d0e765e62be00b8faa074849a8b4207b9c825cda43560965521aa3776e3a1284 |
|
MD5 | 63d47890d2df191ad29c77aa5e4e147a |
|
BLAKE2b-256 | 618458ee1145a9b759583bef85ac3a97ca1e33f2c4e99bc7070bbe8a7a0018db |
Close
Hashes for mlflow_baidubce_store_plugin-1.0.7-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 8a744c84de177e3a3fcc7026333510bd7eda2b5171f808b2f3c86b0605ca307d |
|
MD5 | 0d0bca1d3195cf902c986d21f88cb785 |
|
BLAKE2b-256 | b1b868cb0ba4ece9fb26829d1ba683fde67b6782e29ea486010143f448961191 |