mxxxxxs / learningdl Goto Github PK

for (let j = 0; j < trainingTime; j++) {
for (let i = 0; i < trainingSet.length; i++) {
  let input = trainingSet[i].input
  let target = trainingSet[i].output
    const layersOutput = layers.reduce((preLayer, curLayer) => {
       curLayer.input(preLayer)
       return curLayer.forward()
    }, [input])

    finalLayer.input(layersOutput, target)
    const backward0 = finalLayer.backward()
  
    if (sumLoss.length > 3000)
    sumLoss.shift()
    sumLoss.push(finalLayer.sumLoss())

    layers.reverse().reduce((preD, curLayer) => {
      curLayer.update(preD)
      return curLayer.backward(preD)
    }, backward0)
    layers.reverse()
  }