lib/tensorflow/keras/datasets/mnist.rb in tensorflow-0.1.1 vs lib/tensorflow/keras/datasets/mnist.rb in tensorflow-0.1.2

@@ -1,15 +1,16 @@
 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"
+        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
       end
     end
   end