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

Added hyperparameter optimization script

Browse Source
This commit is contained in:
Victor Mylle
2023-11-27 16:06:05 +00:00
parent c1152ff96c
commit f9e8f9e69f
3 changed files with 106 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/trainers/trainer.py
View File

@@ -78,8 +78,7 @@ class Trainer:
self.data_processor = task.connect(self.data_processor, name="data_processor")
self = task.connect(self, name="trainer")
task.delete_parameter("trainer/quantiles")
task.connect(self.data_processor.data_config, name="data_features")
task.delete_parameter("trainer/quantiles", force=True)
def random_samples(self, train: bool = True, num_samples: int = 10):
train_loader, test_loader = self.data_processor.get_dataloaders(