StandardScaler added

This commit is contained in:
erogol 2020-03-17 13:27:53 +01:00
parent cef9f06887
commit 77f36b65b8
1 changed files with 25 additions and 0 deletions

View File

@ -50,3 +50,28 @@ def pad_per_step(inputs, pad_len):
inputs, [[0, 0], [0, 0], [0, pad_len]],
mode='constant',
constant_values=0.0)
# pylint: disable=attribute-defined-outside-init
class StandardScaler():
def set_stats(self, mean, scale):
self.mean_ = mean
self.scale_ = scale
def reset_stats(self):
delattr(self, 'mean_')
delattr(self, 'scale_')
def transform(self, X):
X = np.asarray(X)
X -= self.mean_
X /= self.scale_
return X
def inverse_transform(self, X):
X = np.asarray(X)
X *= self.scale_
X += self.mean_
return X