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 # get indices of all 00:15 timestamps
if self.full_day_skip: if self.full_day_skip:
start_of_day_indices = dataframe[ 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 ].index
skip_indices.extend(start_of_day_indices) skip_indices.extend(start_of_day_indices)
skip_indices = list(set(skip_indices)) skip_indices = list(set(skip_indices))

228
src/policies/policy_tester.ipynb
View File

File diff suppressed because one or more lines are too long

1
src/trainers/diffusion_trainer.py
View File

@@ -48,6 +48,7 @@ class DiffusionTrainer:
""" """
return torch.randint(low=1, high=self.noise_steps, size=(n,)) return torch.randint(low=1, high=self.noise_steps, size=(n,))
def sample(self, model: DiffusionModel, n: int, inputs: torch.tensor): def sample(self, model: DiffusionModel, n: int, inputs: torch.tensor):
inputs = inputs.repeat(n, 1).to(self.device) inputs = inputs.repeat(n, 1).to(self.device)
model.eval() model.eval()