Merge remote-tracking branch 'origin/master'

This commit is contained in:
steffen 2020-05-15 19:48:07 +02:00
commit 33df856595
2 changed files with 3 additions and 2 deletions

View File

@ -13,8 +13,8 @@ class NoiseInjection(object):
def __call__(self, x: np.ndarray):
if self.noise_factor:
noise = np.random.normal(loc=self.mu, scale=self.sigma, size=x.shape)
augmented_data = x + self.noise_factor * noise
noise = np.random.uniform(0, self.noise_factor, size=x.shape)
augmented_data = x + x * noise
# Cast back to same data type
augmented_data = augmented_data.astype(x.dtype)
return augmented_data

View File

@ -39,6 +39,7 @@ class Config(ConfigParser, ABC):
h = hashlib.md5()
params = deepcopy(self.as_dict)
del params['model']['type']
del params['model']['secondary_type']
del params['data']['worker']
del params['main']
h.update(str(params).encode())