Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- batch_size = 20
- filename_queue = tf.train.string_input_producer(["/tmp/input.csv"], num_epochs=100000)
- reader = tf.TextLineReader(skip_header_lines=0)
- _, csv_row = reader.read(filename_queue)
- record_defaults = [[0.], [0.]]
- col_x, col_label = tf.decode_csv(csv_row, record_defaults=record_defaults)
- min_after_dequeue = 10000
- capacity = min_after_dequeue + 3 * batch_size
- x_batch, label_batch = tf.train.shuffle_batch(
- [col_x, col_label], batch_size=batch_size, capacity=capacity,
- min_after_dequeue=min_after_dequeue)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement