Merge branch 'ml_cleanup' into legacy_calcs

BlauGroup · Jan 6, 2024 · 0fa7cc4 · 0fa7cc4
2 parents d04699d + dd034c9
commit 0fa7cc4
Show file tree

Hide file tree

Showing 19 changed files with 485 additions and 1,626 deletions.
diff --git a/src/NanoParticleTools/machine_learning/__init__.py b/src/NanoParticleTools/machine_learning/__init__.py
@@ -0,0 +1,3 @@
+from NanoParticleTools.machine_learning.data.dataset import NPMCDataset
+from NanoParticleTools.machine_learning.data.processors import SummedWavelengthRangeLabelProcessor
+from NanoParticleTools.machine_learning.data.datamodule import NPMCDataModule
diff --git a/src/NanoParticleTools/machine_learning/models/discretized_graph/data.py b/src/NanoParticleTools/machine_learning/models/discretized_graph/data.py
diff --git a/src/NanoParticleTools/machine_learning/models/discretized_graph/model.py b/src/NanoParticleTools/machine_learning/models/discretized_graph/model.py