Skip to content

Commit

Permalink
Rename dataset back to validation dataset
Browse files Browse the repository at this point in the history
  • Loading branch information
schobbejak committed Jun 10, 2024
1 parent 26c2fbb commit 58c42b1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion epochalyst/pipeline/model/training/torch_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -826,7 +826,7 @@ def get_model_checkpoint_path(self, epoch: int) -> Path:
return Path(f"{self.trained_models_directory}/{self.get_hash()}_checkpoint_{epoch}.pt")


class TrainTestDataset(Dataset[T_co]):
class TrainValidationDataset(Dataset[T_co]):
"""Dataset as a concatenation of multiple datasets.
This class is useful to assemble different existing datasets.
Expand Down

0 comments on commit 58c42b1

Please sign in to comment.