代码之家 › 专栏 › 技术社区 › Roshni Amber

训练、保存和加载tensorflow模型

tensorflow python-3.x

Roshni Amber · 技术社区 · 7 年前

参照 this 为了训练mnist数据集的gan模型,我想保存一个模型并将其还原以进行进一步的预测。在了解了一些 Saving and Importing a Tensorflow Model 我能够保存和恢复输入和输出的一些变量,但是这网络我只能在一些特定的迭代后保存模型,不能预测一些输出。

2 回复 | 直到 7 年前

Siddhant 7 年前

你指的是 this 导游?它非常清楚地解释了如何以所有可能的格式加载和保存tensorflow模型。

Ferraat 7 年前

如果你是ML新手,我建议你先试试Keras,这更容易使用。见 https://keras.io/getting-started/faq/#how-can-i-save-a-keras-model ,几乎可以使用:

model.save('my_model.h5')

将模型保存到磁盘。

model = load_model('my_model.h5')

加载模型并进行预测

推荐文章

ÎÎÎ½Î· ÎÎ®Î¹Î½Î¿Ï · Python lxml.html语法错误:使用lxml find时XPATH的谓词无效

1 年前

Cam · Pandas列表日期到日期时间

1 年前

RASEL MAHMUD · 为什么以及如何在is_even()函数内的IF条件中递归X变量在满足0后递增?[副本]

1 年前

jjkennedy · Pandas文本文件导入:当每个文件中存在多个表时,自动选择1个表

1 年前

LMC · Numpy数组布尔索引以获取包含元素

1 年前

vr8ce · 非成对标记中特定字符的正则表达式

1 年前

Kernel · 如果指定了crs参数,shapefile的geopandas.read_file将出错

1 年前

ShaAnder · 为什么sqllachemy返回的是类而不是字符串

1 年前

sixtytrees · detectron2软件包未安装(没有名为“torch”的模块),但我安装了torch

1 年前

Pernoctador · Python映射可以复制吗?我需要参考地图

1 年前