1

我已经使用自定义损失函数训练了一个关于 Watson Machine learning 的模型，但是当我尝试部署它时，我收到一条错误消息：

Invalid Input data: Unknow loss function:loss

我知道通常在 Keras 中，您可以使用以下方法加载具有自定义损失函数的模型：

load_model('model.h5', custom_object={'loss': loss_function})

我现在正在寻找一种方法来在我的部署中添加我的损失函数的定义。

4

2 回答 2

0

官方支持回复：

WML 目前不支持 Keras 的自定义函数和层。您可以参考此扩展坞中支持的内容：https ://dataplatform.cloud.ibm.com/docs/content/analyze-data/pm_service_supported_frameworks.html?audience=wdp&context= analytics 似乎没有提到关于损失函数的部分这里。我们将更新文档。

日期：2019-01-28

于 2019-03-04T09:22:45.940 回答
0

您应该能够按照本文档中Custom Compnents in TF Models针对 WML描述的方法进行操作

https://dataplatform.cloud.ibm.com/docs/content/analyze-data/ml-custom_libs_tensorflow.html?audience=wdp&context=analytics

具体要求在此处详述：https ://dataplatform.cloud.ibm.com/docs/content/analyze-data/ml-custom_libs_overview.html#requirements

主要思想是您需要为您的部署指定一个运行时，除此之外感谢默认运行时并以指定的方式打包您的自定义组件。希望这可以帮助。

于 2019-01-23T16:08:34.833 回答