Contents

module TensorFlow
  module Keras
    module Datasets
      module MNIST
        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
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
tensorflow-0.1.1	lib/tensorflow/keras/datasets/mnist.rb