Advertisement
Guest User

Untitled

a guest
Feb 19th, 2017
68
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.78 KB | None | 0 0
  1. import numpy as np
  2. import chainer.functions as F
  3. import chainer.links as L
  4. from chainer import Variable,optimizers,Chain
  5.  
  6. class Model(Chain):
  7. def __init__(self):
  8. super(Model, self).__init__(
  9. l1 = L.Linear(2,1),
  10. )
  11. def __call__(self, x):
  12. # h = self.l1(x)
  13. # sigmoid function
  14. h = F.sigmoid(self.l1(x))
  15. return h
  16.  
  17. model = Model()
  18. optimizer = optimizers.MomentumSGD(lr=0.01, momentum=0.9)
  19. optimizer.setup(model)
  20.  
  21. x = Variable(np.array([[0,0],[0,1],[1,0],[1,1]], dtype=np.float32))
  22. t = Variable(np.array([[0],[1],[1],[1]], dtype=np.float32))
  23.  
  24. for i in range(0,3000):
  25. optimizer.zero_grads()
  26. y = model(x)
  27. loss = F.mean_squared_error(y, t)
  28. loss.backward()
  29. optimizer.update()
  30.  
  31. print("loss:",loss.data)
  32.  
  33. print(y.data)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement