Advertisement
Guest User

Untitled

a guest
Jun 27th, 2017
56
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.78 KB | None | 0 0
  1. HIDDEN_LAYERS = 64
  2. X = tf.placeholder("float", [None, STRING_LENGTH, len(CHARS)])
  3. y = tf.placeholder("float", [None, len(CHARS)])
  4. X_seq = tf.unstack(X, STRING_LENGTH, 1)
  5. lstm_cell = tf.contrib.rnn.BasicLSTMCell(HIDDEN_LAYERS)
  6. #sequence of 12 chars to output of 7
  7. outputs, states = tf.contrib.rnn.static_rnn(lstm_cell, X_seq, dtype=tf.float32)
  8. final_output = outputs[-1]
  9. weights = tf.get_variable("weights", [HIDDEN_LAYERS, len(CHARS)], initializer=tf.random_normal_initializer())
  10. biases = tf.get_variable("biases", [len(CHARS)], initializer=tf.random_normal_initializer())
  11. prediction = tf.add(tf.matmul(final_output, weights), biases)
  12. cost = tf.reduce_mean(tf.nn.softmax_cross_entropy_with_logits(logits=prediction, labels=y))
  13. optimizer = tf.train.AdamOptimizer(1e-2)
  14. train_op = optimizer.minimize(cost)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement