Add Graph support in Dataset and Dataloader

2024-10-23 15:04:28 +02:00
parent eb146ea2ea
commit ccc5f5a322
11 changed files with 125 additions and 75 deletions
--- a/pina/data/supervised_dataset.py
+++ b/pina/data/supervised_dataset.py
@@ -6,7 +6,8 @@ from .base_dataset import BaseDataset

 class SupervisedDataset(BaseDataset):
    """
-    This class extends the BaseDataset to handle datasets that consist of input-output pairs.
+    This class extends the BaseDataset to handle datasets that consist of
+    input-output pairs.
    """
    data_type = 'supervised'
    __slots__ = ['input_points', 'output_points']