Skip to content

Commit 8d6dd01

Browse files
Merge pull request BrainJS#311 from BrainJS/rnn-time-step-cross-validate
Rnn time step cross validate
2 parents bee2042 + 52d137e commit 8d6dd01

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

41 files changed

+4657
-1440
lines changed

README.md

+7-1
Original file line numberDiff line numberDiff line change
@@ -295,14 +295,20 @@ With multiple networks you can train in parallel like this:
295295
const crossValidate = new brain.CrossValidate(brain.NeuralNetwork, networkOptions);
296296
crossValidate.train(data, trainingOptions, k); //note k (or KFolds) is optional
297297
const json = crossValidate.toJSON(); // all stats in json as well as neural networks
298-
const net = crossValidate.toNeuralNetwork();
298+
const net = crossValidate.toNeuralNetwork(); // get top performing net out of `crossValidate`
299299

300300

301301
// optionally later
302302
const json = crossValidate.toJSON();
303303
const net = crossValidate.fromJSON(json);
304304
```
305305

306+
Use `CrossValidate` with these classes:
307+
* `brain.NeuralNetwork`
308+
* `brain.RNNTimeStep`
309+
* `brain.LSTMTimeStep`
310+
* `brain.GRUTimeStep`
311+
306312
An example of using cross validate can be found in [examples/cross-validate.js](examples/cross-validate.js)
307313

308314
### Train Stream

0 commit comments

Comments
 (0)