Method __init__

climart/data_transform/transforms.py:76–78

(self, exp_type: str)

    """ Flattens a dict with array's as values into a 1D vector (or 2D with batch dimension)"""

    def __init__(self, exp_type: str):
        super().__init__(exp_type)
        self._out_dim = sum(self.input_dim.values())

    def transform(self, X: Dict[str, np.ndarray]) -> np.ndarray:
        return np.concatenate([X[key].flatten() for key in INPUT_TYPES], axis=0)