Implement Dataset, Dataloader and DataModule class and fix SupervisedSolver

2024-10-16 11:24:37 +02:00
parent b9753c34b2
commit c9304fb9bb
30 changed files with 770 additions and 784 deletions
--- a/pina/init.py
+++ b/pina/init.py
@@ -5,7 +5,8 @@ __all__ = [
    "Plotter",
    "Condition",
    "SamplePointDataset",
-    "SamplePointLoader",
+    "PinaDataModule",
+    "PinaDataLoader"
 ]

 from .meta import *
@@ -15,4 +16,5 @@ from .trainer import Trainer
 from .plotter import Plotter
 from .condition.condition import Condition
 from .data import SamplePointDataset
-from .data import SamplePointLoader
+from .data import PinaDataModule
+from .data import PinaDataLoader