diff options
Diffstat (limited to 'become_yukarin/dataset/dataset.py')
| -rw-r--r-- | become_yukarin/dataset/dataset.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/become_yukarin/dataset/dataset.py b/become_yukarin/dataset/dataset.py index d259734..368073d 100644 --- a/become_yukarin/dataset/dataset.py +++ b/become_yukarin/dataset/dataset.py @@ -189,7 +189,7 @@ class AcousticFeatureNormalizeProcess(BaseDataProcess): self._mean = mean self._var = var - def __call__(self, data: AcousticFeature, test): + def __call__(self, data: AcousticFeature, test=None): f0 = (data.f0 - self._mean.f0) / numpy.sqrt(self._var.f0) f0[~data.voiced] = 0 return AcousticFeature( @@ -206,7 +206,7 @@ class AcousticFeatureDenormalizeProcess(BaseDataProcess): self._mean = mean self._var = var - def __call__(self, data: AcousticFeature, test): + def __call__(self, data: AcousticFeature, test=None): f0 = data.f0 * numpy.sqrt(self._var.f0) + self._mean.f0 f0[~data.voiced] = 0 return AcousticFeature( |