机器学习训练步数过多是否会干扰训练?附TensorFlow相关疑问
关于TensorFlow批次训练重复数据与超步数训练的疑问解答
嘿,针对你提到的两个问题,我给你拆解清楚:
问题1:第899步后TensorFlow会重复训练数据吗?
答案是肯定的。你设置了num_epochs=None,这个参数的作用就是让数据集迭代器无限循环遍历你的训练数据。当第一轮899步(3372149/3751)把所有数据都过一遍之后,TensorFlow会重新从训练数据集的开头读取批次数据,继续执行训练步骤,直到达到你设置的steps=10000为止。
另外注意你代码里设置了shuffle=False,所以重复遍历的时候,数据的顺序和第一轮完全一致;如果打开shuffle,后续迭代会打乱数据顺序再生成批次。
你的代码片段我整理成代码块方便对照:
train_input_fn = tf.estimator.inputs.numpy_input_fn( x={"x": train_features_numpy}, y=train_labels_numpy, batch_size = 3751, num_epochs= 1, shuffle=False ) # Train nn.train(input_fn=train_input_fn)#, steps=10000)
问题2:超步数训练会破坏模型还是仅冗余?
首先明确:不会破坏特征与标签的关联。每一步训练时,模型还是会拿到对应的特征批次和标签批次,训练逻辑是正常的,不会出现特征和标签不匹配的情况。
但这种超步数训练确实是冗余的,甚至可能带来负面影响:
- 如果你的模型已经在现有数据上学到了所有可学习的模式,继续重复训练只会让模型在训练数据上“死记硬背”,也就是过拟合,这会导致模型在测试集上的表现变差。
- 你提到训练损失没下降,这很可能和数据量不足有关:当模型收敛到当前数据集能达到的最优损失后,继续训练步数,损失可能会进入震荡状态,甚至因为过拟合出现轻微上升的情况——多余的步数完全是无效的,甚至有害。
简单来说,当训练数据量固定时,设置超过完整遍历次数的步数,本质就是让模型反复“刷题”,刷到一定程度就会从学习变成死记硬背,对模型性能没有帮助。
内容的提问来源于stack exchange,提问作者yuonger741
相关产品推荐
相关产品推荐

