Advertisement
Guest User

Untitled

a guest
Sep 26th, 2017
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.57 KB | None | 0 0
  1. class DeepNeuralNetwork {
  2.  
  3. let iterations: Int
  4. let learningRate: Double
  5.  
  6. private var layers = [Layer]()
  7.  
  8. init(iterations: Int, learningRate: Double) {
  9. self.iterations = iterations
  10. self.learningRate = learningRate
  11. }
  12.  
  13. func add(layer: Layer) {
  14. layers.append(layer)
  15. }
  16.  
  17. func compile() {
  18. for i in 0..<layers.count {
  19. layers[i].initialize(
  20. previous: i == 0 ? nil : layers[i - 1],
  21. next: i == layers.count - 1 ? nil : layers[i + 1]
  22. )
  23. }
  24. }
  25. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement