Merge pull request #98 from rahulunair/patch-1

(fix): remove depreciated function call
PumasAI · Aug 11, 2022 · 8482290 · 8482290
2 parents 287adee + 83c4c56
commit 8482290
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/mnist_lenet.jl b/examples/mnist_lenet.jl
@@ -2,8 +2,8 @@
 
 import MLDatasets
 function get_data()
-  xtrain, ytrain = MLDatasets.MNIST.traindata(Float32)
-  xtest, ytest = MLDatasets.MNIST.testdata(Float32)
+  xtrain, ytrain = MLDatasets.MNIST(:train)[:]
+  xtest, ytest = MLDatasets.MNIST(:test)[:]
 
   (
     (reshape(xtrain, 28, 28, 1, :), UInt32.(ytrain .+ 1)),