diff --git a/src/data_handling/data_preparation.py b/src/data_handling/data_preparation.py index badd75aa709e108c2516ccf24a3de915c77ca258..db800f5e50146c4f73cde643f154bcb1a5047437 100644 --- a/src/data_handling/data_preparation.py +++ b/src/data_handling/data_preparation.py @@ -249,6 +249,17 @@ class DataPrep(object): else: self.inverse_transform() + def get_transformation_information(self, variable): + try: + mean = self.mean.sel({'variables': variable}).values + except AttributeError: + mean = None + try: + std = self.std.sel({'variables': variable}).values + except AttributeError: + std = None + return mean, std, self._transform_method + def make_history_window(self, dim: str, window: int) -> None: """ This function uses shifts the data window+1 times and returns a xarray which has a new dimension 'window'