def augment_PCA_shift(data, U, s, sigma=0.2): for sample in range(data.shape[0]): data[sample] = illumination_jitter(data[sample], U, s, sigma) data[sample] -= data[sample].min() data[sample] /= data[sample].max() return data