假设训练数据一共10个,train_data = [1,2,...,10]
测试数据也是10个,label = [10,9,...1]
BATCH_SIZE 这个参数,是说每次喂给神经网络,用来学习的数据的个数。(当然,也有对应数量的 label)
BATCH_SIZE 等于5,相当于10个数字分两次喂完,每次完整的投喂称为 epoch,每个 epoch 里投喂一次叫做 step。
比如这10个数据,我打算训练八遍,所以 epoch 就是 [0,1...7]
又因为每个 epoch 要喂两次,才能用完所有数据,所以每个 epoch 有两个 step。
epoch 0 | step 0 | train_data = [4,3,10,7,6] label = [1,4,5,7,8]
epoch 0 | step 1 | train_data = [1,2,5,8,9] label = [2,3,6,9,10]
epoch 1 | step 0 | train_data = [4,3,10,7,6] label = [1,4,5,7,8]
epoch 1 | step 1 | train_data = [1,2,5,8,9] label = [2,3,6,9,10]
epoch 2 | step 0 | train_data = [4,3,10,7,6] label = [1,4,5,7,8]
epoch 2 | step 1 | train_data = [1,2,5,8,9] label = [2,3,6,9,10]
......
epoch 7 | step 0 | train_data = [4,3,10,7,6] label = [1,4,5,7,8]
epoch 7 | step 1 | train_data = [1,2,5,8,9] label = [2,3,6,9,10]