5c8dbdda60bc0290e967c37a55f47ba3e4342882,basenji/dataset.py,SeqDataset,numpy,#SeqDataset#Any#Any#,188

Before Change

    // initialize inputs and outputs
    seqs_1hot = np.zeros((self.num_seqs, self.seq_length, self.seq_depth))
    targets = np.zeros((self.num_seqs, self.target_length, self.num_targets))

    // collect inputs and outputs
    si = 0
    for seq1_1hot, targets1 in dataset:
      if return_inputs:
        seqs_1hot[si,:,:] = seq1_1hot[0,:,:]
      if return_outputs:
        targets[si,:,:] = targets1[0,:,:]
      si += 1
    assert(si == self.num_seqs)

    // return

After Change

    // initialize inputs and outputs
    seqs_1hot = []
    targets = []

    // collect inputs and outputs
    for seq_raw, targets_raw in dataset:
      if return_inputs:
        seq_1hot = seq_raw.numpy().reshape((self.seq_length,-1))
        seqs_1hot.append(seq_1hot)
      if return_outputs:
        targets1 = targets_raw.numpy().reshape((self.target_length,-1))
        targets.append(targets1)

    // make arrays
    seqs_1hot = np.array(seqs_1hot)
    targets = np.array(targets)

Instances