MCPcopy
hub / github.com/BrainJS/brain.js / testPartition

Method testPartition

src/cross-validate.js:21–39  ·  view source on GitHub ↗

* * @param {object} trainOpts * @param {object} trainSet * @param {object} testSet * @returns {void|*}

(trainOpts, trainSet, testSet)

Source from the content-addressed store, hash-verified

19 * @returns {void|*}
20 */
21 testPartition(trainOpts, trainSet, testSet) {
22 let classifier = new this.Classifier(this.options);
23 let beginTrain = Date.now();
24 let trainingStats = classifier.train(trainSet, trainOpts);
25 let beginTest = Date.now();
26 let testStats = classifier.test(testSet);
27 let endTest = Date.now();
28 let stats = Object.assign({}, testStats, {
29 trainTime: beginTest - beginTrain,
30 testTime: endTest - beginTest,
31 iterations: trainingStats.iterations,
32 trainError: trainingStats.error,
33 learningRate: trainOpts.learningRate,
34 hiddenLayers: classifier.hiddenLayers,
35 network: classifier.toJSON()
36 });
37
38 return stats;
39 }
40
41 /**
42 * Randomize array element order in-place.

Callers 3

trainMethod · 0.95
browser.min.jsFile · 0.80
browser.jsFile · 0.80

Calls 3

testMethod · 0.80
trainMethod · 0.45
toJSONMethod · 0.45

Tested by

no test coverage detected