diff --git a/mnist-node/data.js b/mnist-node/data.js index 425be5053..40e8e50fe 100644 --- a/mnist-node/data.js +++ b/mnist-node/data.js @@ -158,7 +158,7 @@ class MnistDataset { labelsIndex = 3; } const size = this.dataset[imagesIndex].length; - tf.util.assert( + assert( this.dataset[labelsIndex].length === size, `Mismatch in the number of images (${size}) and ` + `the number of labels (${this.dataset[labelsIndex].length})`);