<filename>src/tests/test_data.py<gh_stars>0
"""dataset = MNIST(...)
assert len(dataset) == N_train for training and N_test for test
assert that each datapoint has shape [1,28,28] or [728] depending on how you choose to format
assert that all labels are represented"""

<filename>src/tests/test_data.py<gh_stars>0
"""dataset = MNIST(...)
assert len(dataset) == N_train for training and N_test for test
assert that each datapoint has shape [1,28,28] or [728] depending on how you choose to format
assert that all labels are represented"""

dataset = MNIST(...) assert len(dataset) == N_train for training and N_test for test assert that each datapoint has shape [1,28,28] or [728] depending on how you choose to format assert that all labels are represented