def batch_generator(batch_size, sequence_length): while True: x_shape = (batch_size, sequence_length, num_x_signals) x_batch = np.zeros(shape=x_shape, dtype=np.float16) y_shape = (batch_size, sequence_length, num_y_signals) y_batch = np.zeros(shape=y_shape, dtype=np.float16) for i in range(batch_size): x_batch[i] = x_train_scaled[:sequence_length] y_batch[i] = y_train_scaled[:sequence_length] x_batch_1 = x_batch[ :, :, 0:5] x_batch_2 = x_batch[ :, :, 5:12] yield ([x_batch_1, x_batch_2], y_batch)