Skip to main content
 首页 » 编程设计

python之保存 wide and deep tensorflow 模型

2024年10月17日26zhengyun_ustc

我正在玩 Wide and Deep learning example from Tensorflow .我想保存经过训练的分类器以供以后用于预测任务,但我真的不知道该怎么做。 DNNLinearCombinedClassifier 没有可用的保存方法,并且 pickle 对象也失败。

有什么办法可以保存吗?

请您参考如下方法:

this doc的检查点保存部分应该回答你的问题。