Guest User

Untitled

a guest
Dec 10th, 2018
96
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.64 KB | None | 0 0
  1. def backward(reversedWeights: List[Tensor], predictions: List[Tensor], deltas: List[Tensor]): List[Tensor] = {
  2. // the first weighted layer was not influenced by any layer before, so we stop here
  3. if (reversedWeights.length <= 1) deltas
  4. else {
  5. // calculate an error
  6. val error = deltas.head.dot(reversedWeights.head.T)
  7. // calculate an error impact
  8. val delta = error * derivate(predictions.head)
  9. // go for next calculation and prepend a current calculation to a result list
  10. backward(reversedWeights.tail, predictions.tail, delta :: deltas)
  11. }
  12. }
  13.  
  14. val deltas = backward(weights.reverse, prev_predictions, List(final_delta))
Add Comment
Please, Sign In to add comment