electionprocessor.py 文件源码

def next_batch(self):
        df = self.batch_df[self.pointer]
        x = np.array([d[0] for d in df])
        xl = np.array([d[1] for d in df])
        xr = np.array([d[2] for d in df])
        tar = np.array([d[3] for d in df])
        y = np.array([d[-1] for d in df])
        # y = pd.get_dummies(y).values.astype(np.int32)
        seq_len = [len(seq) for seq in x]
        seq_len_l = [len(seq) for seq in xl]
        seq_len_r = [len(seq) for seq in xr]
        if self.dynamic_padding:
            x = np.array(self.pad_minibatches(x, 'RIGHT'))
            xl = np.array(self.pad_minibatches(xl, 'RIGHT'))
            xr = np.array(self.pad_minibatches(xr, 'RIGHT'))
        self.pointer += 1
        return x, y, seq_len, xl, seq_len_l, xr, seq_len_r, tar