Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- lstm_cells = [
- tf.contrib.rnn.LSTMCell(num_units=num_nodes[li],
- state_is_tuple=True,
- initializer= tf.contrib.layers.xavier_initializer()
- )
- for li in range(n_layers)]
- drop_lstm_cells = [tf.contrib.rnn.DropoutWrapper(
- lstm, input_keep_prob=1.0,output_keep_prob=1.0-dropout, state_keep_prob=1.0-dropout
- ) for lstm in lstm_cells]
- drop_multi_cell = tf.contrib.rnn.MultiRNNCell(drop_lstm_cells)
- multi_cell = tf.contrib.rnn.MultiRNNCell(lstm_cells)
- w = tf.get_variable('w',shape=[num_nodes[-1], 1], initializer=tf.contrib.layers.xavier_initializer())
- b = tf.get_variable('b',initializer=tf.random_uniform([1],-0.1,0.1))
Add Comment
Please, Sign In to add comment