请选择 进入手机版 | 继续访问电脑版

大蛇智能

 找回密码
 立即注册

扫一扫,访问微社区

搜索
热搜: 活动 交友 discuz
查看: 154|回复: 3

tensorflow估算器怎样在训练过程中加载验证集

[复制链接]

1

主题

2

帖子

13

积分

新手上路

Rank: 1

积分
13
发表于 2019-11-21 21:49:59 | 显示全部楼层 |阅读模式
tensorflow估算器怎样在训练过程中加载验证集通过在验证集上的准确率来实现停止训练?
回复

使用道具 举报

127

主题

306

帖子

977

积分

管理员

Rank: 9Rank: 9Rank: 9

积分
977
发表于 2019-11-22 09:32:01 | 显示全部楼层
加入验证集可以使用tf.estimator.train_and_evaluate 函数。具体用法见实例26 和27.
如果加入早停功能可以将tf.estimator.train_and_evaluate拆开。用estimator.train和estimator.evaluate组合来做
回复

使用道具 举报

1

主题

2

帖子

13

积分

新手上路

Rank: 1

积分
13
 楼主| 发表于 2019-11-22 10:54:54 | 显示全部楼层
admin 发表于 2019-11-22 09:32
加入验证集可以使用tf.estimator.train_and_evaluate 函数。具体用法见实例26 和27.
如果加入早停功能可以 ...

大神,estimator.train和estimator.evaluate组合,是一直等着train训练结束,evaluate才开始跑,能不能例如总共训练1000步,每跑100步,就在测试集上跑一次,然后看准确率是否有提升,超过多少步没有提升就停止训练。
回复

使用道具 举报

127

主题

306

帖子

977

积分

管理员

Rank: 9Rank: 9Rank: 9

积分
977
发表于 2020-2-2 22:57:38 | 显示全部楼层
估算器本来就是高度集成化工具。为了使用方便,所以将接口能力做得很弱。如果要求细粒度控制,可以直接实现训练过程即可。另外可以使用keras的fit方式来训练和估算器很像。你说的功能它都有
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

Archiver|手机版|小黑屋|大蛇智能 ( 京ICP备 18026897 号 )

GMT+8, 2020-2-21 17:06 , Processed in 0.033919 second(s), 23 queries .

Powered by Discuz! X3.4

© 2001-2017 Comsenz Inc.

快速回复 返回顶部 返回列表