数据集无法完全训练怎么办?

新手上路,请多包涵

训练集和测试集大小如下:
print(x_train.shape)
print(y_train.shape)
print(x_test.shape)
print(y_test.shape)
(12724, 64, 64, 3)
(12724, 5)
(3182, 64, 64, 3)
(3182, 5)
但是在实际训练在只能用到部分数据
image.png
请问各位这个问题怎么解决

阅读 1.9k
1 个回答

12724 / 32 = 397.625
ceil(12724 / 32) = 398

所以这个显示会不会不是样本数而是 batch 数