add dataset and dataloader for sample points (#195)

* add dataset and dataloader for sample points * unittests
2023-11-07 11:34:44 +01:00
parent cd5bc9a558
commit d654259428
19 changed files with 581 additions and 196 deletions
--- a/tests/test_label_tensor.py
+++ b/tests/test_label_tensor.py
@@ -95,10 +95,14 @@ def test_getitem():
 def test_getitem2():
    tensor = LabelTensor(data, labels)
    tensor_view = tensor[:5]
-
    assert tensor_view.labels == labels
    assert torch.allclose(tensor_view, data[:5])

+    idx = torch.randperm(tensor.shape[0])
+    tensor_view = tensor[idx]
+    assert tensor_view.labels == labels
+
+
 def test_slice():
    tensor = LabelTensor(data, labels)
    tensor_view = tensor[:5, :2]