diff --git a/LGHackerton/models/patchtst_trainer.py b/LGHackerton/models/patchtst_trainer.py index f0665a5..b44a5d3 100644 --- a/LGHackerton/models/patchtst_trainer.py +++ b/LGHackerton/models/patchtst_trainer.py @@ -156,7 +156,7 @@ def _make_rocv_slices( n = len(label_dates) slices: List[Tuple[np.ndarray, np.ndarray]] = [] - used = np.zeros(n, dtype=bool) + used = np.zeros(n, dtype=bool) # prevents overlap among folds dates = np.sort(np.unique(label_dates)) for i in range(n_folds):