【Python学习】 - TensorFlow.keras 不显示epochs进度条的方法

一、概述

在我们使用TensorFlow进行神经网络的搭建时,难免遇到需要训练很多次来拟合数据的情况,假设需要拟合1000次数据,那么可能前800次的拟合效果都不是很好,所以显示进度条就会使得输出面板被填满,输出的信息我们并不关心,我们只关心最后200次的拟合效果,此时思考能否可以有一种办法可以简便的在训练多个epochs时隐藏进度条的输出呢?

 

二、具体操作

阅读这个函数

tensorflow.keras.models.Sequential.fit

在上述函数原型中,发现在对参数介绍的时候有这样一个参数:verbose。

verbose: 0, 1, or 2. Verbosity mode.
            0 = silent, 1 = progress bar, 2 = one line per epoch.
            Note that the progress bar is not particularly useful when
            logged to a file, so verbose=2 is recommended when not running
            interactively (eg, in a production environment).

翻译:

verbose:日志显示,0为不在标准输出流输出日志信息,1为输出进度条记录,2为每个epoch输出一行记录。

 

所以运行代码的时候我们只需要令这个变量=0即可取消进度条的显示。该变量默认=1

model.fit(X_train,y_train,epochs = 1000,verbose=0)

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值