Merge branch '310-fix-gpu-to-json' into rnn-time-step-cross-validate

sygops · Nov 26, 2018 · 3b66cbc · 3b66cbc
2 parents ef269fc + 5614561
commit 3b66cbc
Show file tree

Hide file tree

Showing 10 changed files with 577 additions and 1,020 deletions.
diff --git a/README.md b/README.md
@@ -354,6 +354,7 @@ The output of `train()` is a hash of information about how the training went:
 Supported on classes:
 
 * `brain.NeuralNetwork`
+* `brain.NeuralNetworkGPU` -> All the functionality of `brain.NeuralNetwork` but, ran on GPU (via gpu.js in WebGL2, WebGL1, or fallback to CPU)
 * `brain.recurrent.RNN`
 * `brain.recurrent.LSTM`
 * `brain.recurrent.GRU`
@@ -503,6 +504,7 @@ Likely example see: [simple letter detection](./examples/which-letter-simple.js)
 
 # Neural Network Types
 * [`brain.NeuralNetwork`](src/neural-network.js) - [Feedforward Neural Network](https://en.wikipedia.org/wiki/Feedforward_neural_network) with backpropagation
+* [`brain.NeuralNetworkGPU`](src/neural-network-gpu.js) - [Feedforward Neural Network](https://en.wikipedia.org/wiki/Feedforward_neural_network) with backpropagation, GPU version
 * [`brain.recurrent.RNNTimeStep`](src/recurrent/rnn-time-step.js) - [Time Step Recurrent Neural Network or "RNN"](https://en.wikipedia.org/wiki/Recurrent_neural_network)
 * [`brain.recurrent.LSTMTimeStep`](src/recurrent/lstm-time-step.js) - [Time Step Long Short Term Memory Neural Network or "LSTM"](https://en.wikipedia.org/wiki/Long_short-term_memory)
 * [`brain.recurrent.GRUTimeStep`](src/recurrent/gru-time-step.js) - [Time Step Gated Recurrent Unit or "GRU"](https://en.wikipedia.org/wiki/Gated_recurrent_unit)