N_BATCHES = 1 logger.info("Preparing synthetic mains data for {} batches.".format(N_BATCHES)) mains = None targets = None TARGET_I = 4 for batch_i in range(N_BATCHES): batch = mains_source.queue.get(timeout=30) mains_batch, targets_batch = batch.data if mains is None: mains = mains_batch targets = targets_batch[:, :, TARGET_I] else: mains = np.concatenate((mains, mains_batch)) targets = np.concatenate((targets, targets_batch[:, :, TARGET_I])) mains_source.stop() # Post-process data seq_length = net.input_shape[1] def pad(data): return np.pad(data, (seq_length, seq_length), mode='constant', constant_values=(data.min().astype(float), )) mains = pad(mains.flatten()) targets = pad(targets.flatten()) logger.info("Done preparing synthetic mains data!")
N_BATCHES = 1 logger.info("Preparing synthetic mains data for {} batches.".format(N_BATCHES)) mains = None targets = None TARGET_I = 2 for batch_i in range(N_BATCHES): batch = mains_source.queue.get(timeout=30) mains_batch, targets_batch = batch.data if mains is None: mains = mains_batch targets = targets_batch[:, :, TARGET_I] else: mains = np.concatenate((mains, mains_batch)) targets = np.concatenate((targets, targets_batch[:, :, TARGET_I])) mains_source.stop() # Post-process data seq_length = net.input_shape[1] def pad(data): return np.pad(data, (seq_length, seq_length), mode='constant', constant_values=(data.min().astype(float), )) mains = pad(mains.flatten()) targets = pad(targets.flatten()) logger.info("Done preparing synthetic mains data!") # Unstandardise for plotting