()
| 907 | } |
| 908 | |
| 909 | function oneStep(): void { |
| 910 | iter++; |
| 911 | trainData.forEach((point, i) => { |
| 912 | let input = constructInput(point.x, point.y); |
| 913 | nn.forwardProp(network, input); |
| 914 | nn.backProp(network, point.label, nn.Errors.SQUARE); |
| 915 | if ((i + 1) % state.batchSize === 0) { |
| 916 | nn.updateWeights(network, state.learningRate, state.regularizationRate); |
| 917 | } |
| 918 | }); |
| 919 | // Compute the loss. |
| 920 | lossTrain = getLoss(network, trainData); |
| 921 | lossTest = getLoss(network, testData); |
| 922 | updateUI(); |
| 923 | } |
| 924 | |
| 925 | export function getOutputWeights(network: nn.Node[][]): number[] { |
| 926 | let weights: number[] = []; |
no test coverage detected
searching dependent graphs…