Method sample

snake-dqn/replay_memory.js:61–73

* Randomly sample a batch of items from the replay buffer. * * The sampling is done *without* replacement. * * @param {number} batchSize Size of the batch. * @return {Array } Sampled items.

(batchSize)

   * @return {Array<any>} Sampled items.
   */
  sample(batchSize) {
    if (batchSize > this.maxLen) {
      throw new Error(
          `batchSize (${batchSize}) exceeds buffer length (${this.maxLen})`);
    }
    tf.util.shuffle(this.bufferIndices_);

    const out = [];
    for (let i = 0; i < batchSize; ++i) {
      out.push(this.buffer[this.bufferIndices_[i]]);
    }
    return out;
  }
}