(batchSize)
| 106 | } |
| 107 | |
| 108 | nextTrainBatch(batchSize) { |
| 109 | return this.nextBatch( |
| 110 | batchSize, [this.trainImages, this.trainLabels], () => { |
| 111 | this.shuffledTrainIndex = |
| 112 | (this.shuffledTrainIndex + 1) % this.trainIndices.length; |
| 113 | return this.trainIndices[this.shuffledTrainIndex]; |
| 114 | }); |
| 115 | } |
| 116 | |
| 117 | nextTestBatch(batchSize) { |
| 118 | return this.nextBatch(batchSize, [this.testImages, this.testLabels], () => { |