module TensorFlow::Keras::Datasets::MNIST

Public Class Methods

load_data(path: "mnist.npz") click to toggle source
# File lib/tensorflow/keras/datasets/mnist.rb, line 5
def self.load_data(path: "mnist.npz")
  file = Utils.get_file(
    path,
    "https://storage.googleapis.com/tensorflow/tf-keras-datasets/mnist.npz",
    file_hash: "731c5ac602752760c8e48fbffcf8c3b850d9dc2a2aedcf2cc48468fc17b673d1"
  )

  data = Npy.load_npz(file)
  [[data["x_train"], data["y_train"]], [data["x_test"], data["y_test"]]]
end