Method loadData

quantization/data_mnist.js:180–196

Loads training and test data.

()

  /** Loads training and test data. */
  async loadData() {
    const baseUrlAndFilePaths = this.getBaseUrlAndFilePaths();
    const baseUrl = baseUrlAndFilePaths.baseUrl;
    const destDir = baseUrlAndFilePaths.destDir;
    if (!(await exists(destDir))) {
      await mkdir(destDir);
    }

    this.dataset = await Promise.all([
      loadImages(baseUrl, destDir, baseUrlAndFilePaths.trainImages),
      loadLabels(baseUrl, destDir, baseUrlAndFilePaths.trainLabels),
      loadImages(baseUrl, destDir, baseUrlAndFilePaths.testImages),
      loadLabels(baseUrl, destDir, baseUrlAndFilePaths.testLabels)
    ]);
    this.trainSize = this.dataset[0].length;
    this.testSize = this.dataset[2].length;
  }

  getTrainData() {
    return this.getData_(true);