28f6dbec4bee2572fa7f94445d63cebb2de6dc9b,bin/tfr_hdf5.py,,read_tfr,#Any#Any#,93

Before Change

  iterator = dataset.make_one_shot_iterator()

  next_op = iterator.get_next()

  seqs_1hot = []
  targets = []

  with tf.Session() as sess:
    next_datum = sess.run(next_op)
    while next_datum:
      seq_1hot = next_datum["sequence"].reshape((-1,4))
      targets1 = next_datum["targets"].reshape(target_len,-1)

After Change

  for seq_1hot, targets1 in dataset:
    // TEMP!
    if si % 2 == 0:
      seq_1hot = seq_1hot.numpy()[0].astype("uint8")
      targets1 = targets1.numpy()[0].astype("float16")
      seq_1hot = seq_1hot.reshape((-1,4))
      targets1 = targets1.reshape((-1,num_targets))
      seqs_1hot.append(seq_1hot)

Instances