Fix readme

r2dtools · Jul 1, 2023 · d0a7ceb · d0a7ceb
1 parent be64c25
commit d0a7ceb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -49,10 +49,10 @@ if err != nil {
     panic(err)
 }
 
-images := dense.NewImageDenses(train.Images, 1, 1000) // convert images data to a slice of gonum matrices with dimension 1x1000
+images := dense.NewImageDenses(train.Images, 1, loader.ImageWidth*loader.ImageHeight) // convert images data to a slice of gonum matrices with dimension 1x784
 labels := dense.NewLabelVecDense(train.Labels)
 
-normalizedImages := dense.NewNormalizedImageDenses(train.Images, 1, 1000, 255) // convert images data to a slice of gonum matrices with dimension 1x1000. Devide all element by 255
+normalizedImages := dense.NewNormalizedImageDenses(train.Images, 1, loader.ImageWidth*loader.ImageHeight, 255) // convert images data to a slice of gonum matrices with dimension 1x784. Devide all element by 255
 
 ....
 ```