keras中deserialize_keras_object的使用

deserialize_keras_object函数的输入格式可以为dict、string。
1、损失函数是自定义的时(字符串),如’rank_hinge_loss’,写上定义函数,module_objects=globals()
2、损失函数是Keras自带的时,‘categorical_crossentropy’,module_objects=globals()
3、优化函数有多个时,用dict格式,module_objects=all_dict

猜你喜欢

转载自blog.csdn.net/angel_hben/article/details/88173439
今日推荐