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.6.tar.gz
Algorithm | Hash digest | |
---|---|---|
SHA256 | 393281bbf35040ca9930a5bc45bdbb84fb152c7867963188611cf8b4f925d161 |
|
MD5 | 3ce28f70266416e4854eea7939220f9d |
|
BLAKE2b-256 | 861b6b9745ce238ded2d6c749ea0a141fe5cdd112dccebf9ce3d6cfa92faf5d9 |
Close
Hashes for mlflow_baidubce_store_plugin-1.0.6-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 5bd52cc58ab7c0f5fed78b0557a4960da9d25317881c124f9e270d3ec94ea579 |
|
MD5 | 2279ffe7af369ff8eb3459bae4dea01a |
|
BLAKE2b-256 | 5c7aad8fa27b6880c01dbfb36ad2e453eed76c4a71583a8742a3ddbc7c1c8ed2 |