diff --git a/pipe/etl.py b/pipe/etl.py index d83e4df..854f242 100644 --- a/pipe/etl.py +++ b/pipe/etl.py @@ -85,7 +85,7 @@ def split_sets(data: DataFrame, split=[0.99, 0.005, 0.005]) -> tuple: def trim(dataframe, column): ndarray = np.array(dataframe.select(column).collect()) \ - .reshape(-1, 32, 133) + .reshape(-1, 34, 133) return ndarray diff --git a/pipe/extract.py b/pipe/extract.py index 4d2c1e6..40c650a 100644 --- a/pipe/extract.py +++ b/pipe/extract.py @@ -105,7 +105,7 @@ def metadata_read(self, metapath: Path, labels:list, def read_matfiles(self, specpath: Path, datakinds: List[DataKind], - default_size: tuple = (32, 130), + default_size: tuple = (34, 130), pad_value: float = 0.) -> DataFrame: """ Loads data for each stack iteration from a set of mat files,