Merge pull request #153 from BrainJS/152-NaN-error-rate

fix #152 and bump version number
BrainJS · Feb 11, 2018 · cb9a707 · cb9a707
2 parents 6df3f90 + 4009d95
commit cb9a707
Show file tree

Hide file tree

Showing 7 changed files with 143 additions and 213 deletions.
diff --git a/bower.json b/bower.json
@@ -31,5 +31,5 @@
     "node_modules",
     "test"
   ],
-  "version": "1.0.4"
+  "version": "1.0.5"
 }
diff --git a/browser.js b/browser.js
diff --git a/browser.min.js b/browser.min.js
diff --git a/dist/neural-network-gpu.js b/dist/neural-network-gpu.js
diff --git a/dist/neural-network-gpu.js.map b/dist/neural-network-gpu.js.map
diff --git a/package.json b/package.json
@@ -1,7 +1,7 @@
 {
   "name": "brain.js",
   "description": "Neural network library",
-  "version": "1.0.4",
+  "version": "1.0.5",
   "author": "Heather Arthur <[email protected]>",
   "repository": {
     "type": "git",

diff --git a/src/neural-network-gpu.js b/src/neural-network-gpu.js
@@ -236,7 +236,7 @@ export default class NeuralNetworkGPU extends NeuralNetwork {
     this.getMSE = this.gpu.createKernel(mse, {
       output: [1],
       constants: {
-        size: this.outputLayer
+        size: this.sizes[this.outputLayer]
       }
     });
   }