Keras深度学习:入门、实战与进阶
上QQ阅读APP看书,第一时间看更新

3.5.3 使用YAML格式保存模型

和之前JSON格式类似,只是文件格式变为YAML,使用的函数变为model_to_yaml()和model_from_yaml()。

> yaml_config <- model_to_yaml(model)
> writeLines(yaml_config,'models/model_config.yaml')
> # 将网络结构和权重值加载到新模型中
> rm(list='yaml_config')
> yaml_config <- readLines('models/model_config.yaml')
> new_model3 <- model_from_json(json_config)
> load_model_weights_hdf5(new_model3,'models/model_weights.h5')
> predict(new_model3,u)
          [,1]
[1,] 0.431296
> # 重新训练模型
> new_model3 %>% fit(X,y)
Error in py_call_impl(callable, dots$args, dots$keywords) :
    RuntimeError: You must compile your model before training/testing. Use `model.compile(optimizer, loss)`.