feat: graph data sets

src/deep_neurographs/machine_learning/datasets.py

@@ -4,7 +4,7 @@
 @author: Anna Grim
 @email: anna.grim@alleninstitute.org
 
-Custom datasets for deep learning models.
+Custom datasets for training deep learning models.
 
 """
 
@@ -230,11 +230,6 @@ def __getitem__(self, idx):
         return {"inputs": inputs, "targets": self.targets[idx]}
 
 
-class ProposalGraphDataset(Dataset):
-    def __init__(self, neurograph, inputs, labels):
-        pass
-
-
 # Augmentation
 class AugmentImages:
     """
@@ -284,6 +279,13 @@ def run(self, arr):
         return self.transform(arr)
 
 
+def get_lengths(neurograph):
+    lengths = []
+    for edge in neurograph.proposals.keys():
+        lengths.append(neurograph.proposal_length(edge))
+    return lengths
+
+
 # -- utils --
 def reformat(arr):
     """