VictorMylle/Thesis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added diffusion validation set

Browse Source
This commit is contained in:
Victor Mylle
2024-05-17 16:11:17 +00:00
parent 11ae0e1949
commit 8a219d0d19
24 changed files with 64 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/data/preprocessing.py
View File

@@ -50,8 +50,8 @@ class DataProcessor:
datetime(year=2022, month=11, day=30, tzinfo=pytz.UTC),
)
self.val_range = (
datetime(year=2022, month=10, day=1, tzinfo=pytz.UTC),
datetime(year=2022, month=11, day=30, tzinfo=pytz.UTC),
datetime(year=2022, month=11, day=1, tzinfo=pytz.UTC),
datetime(year=2022, month=12, day=30, tzinfo=pytz.UTC),
)
self.test_range = (datetime(year=2023, month=1, day=1, tzinfo=pytz.UTC), np.inf)
@@ -297,10 +297,10 @@ class DataProcessor:
):
val_df = self.all_features.copy()
if self.test_range[0] != -np.inf:
if self.val_range[0] != -np.inf:
val_df = val_df[(val_df["datetime"] >= self.val_range[0])]
if self.test_range[1] != np.inf:
if self.val_range[1] != np.inf:
val_df = val_df[(val_df["datetime"] <= self.val_range[1])]
if transform: