上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)`.