微信公众号搜"智元新知"关注
微信扫一扫可直接关注哦!

如何将部署的 ML 模型下载/导出到我的本地机器?

如何解决如何将部署的 ML 模型下载/导出到我的本地机器?

使用 IBM Watson,我可以使用其 Auto AI 平台或 Jupyter Notebook 创建和部署 ML 模型。一旦部署了我的模型,那么我如何下载/导出完成预测的 ML 模型,即如何获得最终的 sklearn 模型(.pkl 文件)、Tensorflow 模型(.pb 文件)或其他支持的 ML 模型在我的本地机器上?

解决方法

您可以使用以下 WML API 调用来下载模型的内容:-

curl -X PUT 'https://us-south.ml.cloud.ibm.com/ml/v4/models/:model_id/content?content_format=<string>&space_id=<string>&project_id=<string>&pipeline_node_id=<string>&name=<string>&version=2020-09-01' --data-raw '"<object>"'

https://cloud.ibm.com/apidocs/machine-learning#models-download-content

您也可以使用 WML Python 客户端 https://wml-api-pyclient-dev-v4.mybluemix.net/#repository

client.repository.download(model_uid,'my_model.tar.gz')

或者,如果您现在正在使用部署空间,您可以简单地导出空间并选择您想要导出的模型。 在导出的 zip 文件中,您将在 assets/wml_model/ 目录下找到模型文件。

版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。