(fix): remove depreciated function call

MNIST.traindata() is deprecated, change fixes code to use `MNIST(split=:train)[:]` instead.
PumasAI · Aug 10, 2022 · 83c4c56 · 83c4c56
1 parent 287adee
commit 83c4c56
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)),