如何使Keras自动加载自定义指标

时间:2018-08-06 18:19:00

标签: python tensorflow keras root-framework

我正在将Keras与Cern ROOT及其分析软件包TMVA一起使用。它的工作方式是我使用Keras初始化NN,然后将其保存到文件中,然后TMVA加载该文件。问题是我在设置神经网络时正在使用自定义指标,而在这样做时,Keras希望你做类似的事情

    models.load_model(model_path, custom_objects={"my_object":my_object})

不幸的是,TMVA接受参数的方式要求我仅提供所用模型文件的文件名。但是,根据收到的错误消息,很明显,它只是使用Keras加载模型。我的问题是,如何强制Keras自动加载我的自定义对象,而不必使用上面的行,因为它与我要使用的软件包不兼容。

0 个答案:

没有答案