Add Mental Images Examples

docs/src/examples.md

@@ -34,6 +34,28 @@ s = count(classify(wnn, test_x) .== test_y)
 println("Accuracy: $(100. * s / 10_000)%")
 ```
 
+```@example mnist
+using Images
+
+images = WiSARD.images(wnn)
+
+img = images[0]
+
+RGB.([img[(i - 1) * 28 + j] for i = 1:28, j = 1:28])
+```
+
+```@example mnist
+img = images[1]
+
+RGB.([img[(i - 1) * 28 + j] for i = 1:28, j = 1:28])
+```
+
+```@example mnist
+img = images[2]
+
+RGB.([img[(i - 1) * 28 + j] for i = 1:28, j = 1:28])
+```
+
 ## Fashion MNIST
 ```@example fashion-mnist
 using WiSARD
@@ -60,4 +82,26 @@ train!(wnn, train_y, train_x)
 s = count(classify(wnn, test_x) .== test_y)
 
 println("Accuracy: $(100. * s / 10_000)%")
+```
+
+```@example fashion-mnist
+using Images
+
+images = WiSARD.images(wnn)
+
+img = images[0]
+
+RGB.([img[(i - 1) * 28 + j] for i = 1:28, j = 1:28])
+```
+
+```@example fashion-mnist
+img = images[1]
+
+RGB.([img[(i - 1) * 28 + j] for i = 1:28, j = 1:28])
+```
+
+```@example fashion-mnist
+img = images[2]
+
+RGB.([img[(i - 1) * 28 + j] for i = 1:28, j = 1:28])
 ```

src/images/images.jl

@@ -19,10 +19,10 @@ function images(wnn::WNN{S, T}, y::S) where {S, T}
     return if M == 0
         Float64.(img)
     else
-        Float64.(img / M)
+        Float64.(img ./ M)
     end
 end
 
-function images(wnn::WNN)
-    return Dict{S, Array{Int}}(y => images(wnn, y) for y in keys(wnn.cls))
+function images(wnn::WNN{S, T}) where {S, T}
+    return Dict{S, Array{Float64}}(y => images(wnn, y) for y in keys(wnn.cls))
 end