Advertisement
Guest User

Untitled

a guest
Aug 16th, 2017
49
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.52 KB | None | 0 0
  1. batch_size = 20
  2. filename_queue = tf.train.string_input_producer(["/tmp/input.csv"], num_epochs=100000)
  3. reader = tf.TextLineReader(skip_header_lines=0)
  4. _, csv_row = reader.read(filename_queue)
  5. record_defaults = [[0.], [0.]]
  6. col_x, col_label = tf.decode_csv(csv_row, record_defaults=record_defaults)
  7. min_after_dequeue = 10000
  8. capacity = min_after_dequeue + 3 * batch_size
  9. x_batch, label_batch = tf.train.shuffle_batch(
  10. [col_x, col_label], batch_size=batch_size, capacity=capacity,
  11. min_after_dequeue=min_after_dequeue)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement