Function loadLabels

quantization/data_mnist.js:128–155

(baseURL, destDir, filename)

}

async function loadLabels(baseURL, destDir, filename) {
  const buffer =
      await fetchOnceAndSaveToDiskWithBuffer(baseURL, destDir, filename);

  const headerBytes = LABEL_HEADER_BYTES;
  const recordBytes = LABEL_RECORD_BYTE;

  const headerValues = loadHeaderValues(buffer, headerBytes);
  tf.util.assert(
      headerValues[0] === LABEL_HEADER_MAGIC_NUM,
      () => `Label file header doesn't match expected magic num.`);

  const labels = [];
  let index = headerBytes;
  while (index < buffer.byteLength) {
    const array = new Int32Array(recordBytes);
    for (let i = 0; i < recordBytes; i++) {
      array[i] = buffer.readUInt8(index++);
    }
    labels.push(array);
  }

  tf.util.assert(
      labels.length === headerValues[1],
      () => `Actual labels length (${images.length} doesn't match ` +
      `value in header (${headerValues[1]})`);
  return labels;
}

/** Helper class to handle loading training and test data. */
export class MnistDataset {