Merge pull request #54 from BlauGroup/legacy_calcs

Fixed and Updated GNN model
BlauGroup · Jan 6, 2024 · fb4b5ec · fb4b5ec
2 parents 6bc07a9 + 5867e5f
commit fb4b5ec
Show file tree

Hide file tree

Showing 19 changed files with 484 additions and 1,633 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