35000f76437852e63287719d948723c322e2327e,basenji/batcher.py,Batcher,next,#Batcher#,26

Before Change

            // copy data
            Xb = np.array(self.Xf[self.start:stop], dtype="float32")
            if self.Yf is not None:
                Yb = np.nan_to_num(np.array(self.Yf[self.start:stop], dtype="float32"))

            // specify full batch
            Nb = self.batch_size

After Change

                // fix N positions
                Xbi_n = (Xb[i].sum(axis=1) == 0)
                Xb[i] = Xb[i] + (1/self.seq_depth)*Xbi_n.repeat(self.seq_depth).reshape(self.seq_len,self.seq_depth)

                if self.Yf is not None:
                    Yb[i] = np.nan_to_num(self.Yf[si])

Instances