* * @param {object} trainOpts * @param {object} trainSet * @param {object} testSet * @returns {void|*}
(trainOpts, trainSet, testSet)
| 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. |
no test coverage detected