Guest User

Untitled

a guest
Mar 21st, 2018
78
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.94 KB | None | 0 0
  1. def generate_data():
  2. X=[]
  3. Y=[]
  4. for i in range(50000):
  5. start=random.randint(1,100)
  6. d=random.randrange(-1,2,2) #-1 or 1
  7. param=[(start),(start+d),(start+d+d)]
  8. X.append(np.array(param))
  9. if d<0:
  10. Y.append([1,0])
  11.  
  12. elif len(Y)>2 and d>0 and Y[-1][1]==1 and Y[-2][1]==1:
  13. Y.append([1,0])
  14. elif d>0:
  15. Y.append([0,1])
  16. X=np.array(X)
  17. Y=np.array(Y)
  18. return X,Y
  19. X,Y = generate_data()
  20. X=np.asarray(X,'float32')
  21. Y=np.asarray(Y,'float32')
  22. X=np.reshape(X,(1,len(X),3))
  23. Y=np.reshape(Y,(1,len(Y),2))
  24.  
  25. model=Sequential()
  26. model.add(LSTM(20, input_shape=(50000,3), return_sequences=True))
  27. model.add(Dense(2))
  28. model.add(Activation('softmax'))
  29.  
  30. model.compile(loss='categorical_crossentropy', optimizer=RMSprop(), metrics=['accuracy'])
  31. history = model.fit(X, Y,batch_size=100, nb_epoch=250, verbose=2)
  32.  
  33. model.add(CuDNNLSTM(20, input_shape=(50000,3), return_sequences=True))
Add Comment
Please, Sign In to add comment