Guest User

Untitled

a guest
May 23rd, 2018
80
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.69 KB | None | 0 0
  1. lstm_cells = [
  2. tf.contrib.rnn.LSTMCell(num_units=num_nodes[li],
  3. state_is_tuple=True,
  4. initializer= tf.contrib.layers.xavier_initializer()
  5. )
  6. for li in range(n_layers)]
  7.  
  8. drop_lstm_cells = [tf.contrib.rnn.DropoutWrapper(
  9. lstm, input_keep_prob=1.0,output_keep_prob=1.0-dropout, state_keep_prob=1.0-dropout
  10. ) for lstm in lstm_cells]
  11. drop_multi_cell = tf.contrib.rnn.MultiRNNCell(drop_lstm_cells)
  12. multi_cell = tf.contrib.rnn.MultiRNNCell(lstm_cells)
  13.  
  14. w = tf.get_variable('w',shape=[num_nodes[-1], 1], initializer=tf.contrib.layers.xavier_initializer())
  15. b = tf.get_variable('b',initializer=tf.random_uniform([1],-0.1,0.1))
Add Comment
Please, Sign In to add comment