ds shape printout

francesco-vaselli · May 22, 2024 · 19958ed · 19958ed
1 parent de4da1b
commit 19958ed
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/models/new_train_model.py b/src/models/new_train_model.py
@@ -224,6 +224,7 @@ def train(
     # 1. Load the data
     ds = np.load(data_path)
     ds = filter_stationary_sequences_dataset(ds)
+    print("ds shape:", ds.shape)
     train_x = ds[:n_train, :7]
     train_y = ds[:n_train, 7:]
     val_x = ds[n_train : n_train + n_val, :7]