美文网首页深度学习-推荐系统-CV-NLP
keras 保存和加载权重及模型结构

keras 保存和加载权重及模型结构

作者: 春二一 | 来源:发表于2019-11-22 15:41 被阅读0次

    模型结构可以很容易地被保存和加载,如下:

    保存为JSON字串 json_string= model. to_json()
    保存为YAML字串 yaml_string=model.to_yaml() to_yam1()
    从JsoN字串重构模

    from keras.models import model_from_json
    model= model_from_json(json_string)
    

    从YAML字串重构模型

    model= model_from_yaml(yaml_string)
    

    模型参数(权重)也可以很容易地被保存和加载,如下

    from keras.models import load_model model. save('my model. h5 ')
    

    创建HDF5文件' my_model.h5
    删除存在的模型
    返回编译好的模型
    和前一模型相同,等于load_model(' my_ model.h5")

    相关文章

      网友评论

        本文标题:keras 保存和加载权重及模型结构

        本文链接:https://www.haomeiwen.com/subject/pwwclctx.html