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

Tried policy with diffusion model

Browse Source
This commit is contained in:
Victor Mylle
2023-12-29 12:30:30 +00:00
parent da3ab3d5b3
commit ef8b5f49ac
3 changed files with 113 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/data/dataset.py
View File

@@ -94,7 +94,7 @@ class NrvDataset(Dataset):
# get indices of all 00:15 timestamps
if self.full_day_skip:
start_of_day_indices = dataframe[
dataframe["datetime"].dt.time != pd.Timestamp("00:15:00").time()
dataframe["datetime"].dt.time != pd.Timestamp("00:00:00").time()
].index
skip_indices.extend(start_of_day_indices)
skip_indices = list(set(skip_indices))