Module: TensorFlow::Keras::Datasets::MNIST

Defined in:
lib/tensorflow/keras/datasets/mnist.rb

Class Method Summary collapse

Class Method Details

.load_data ⇒ Object 



5
6
7
8
9
10
11
12
13
 
# File 'lib/tensorflow/keras/datasets/mnist.rb', line 5

def self.load_data
  # TODO verify hash
  data = Utils.load_dataset(
    "mnist.npz",
    "https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz"
  )

  [[data["x_train"], data["y_train"]], [data["x_test"], data["y_test"]]]
end