Load a buffer with the MNIST dataset. If the data doesn't exist download it and save it for the next time. Args: root (Path or str, optional): The directory to load/save the data. If none is given the ``~/.cache/mlx.data/mnist`` is used. train (bool): Load the t
(root=None, train=True)
| 68 | |
| 69 | |
| 70 | def load_mnist(root=None, train=True): |
| 71 | """Load a buffer with the MNIST dataset. |
| 72 | |
| 73 | If the data doesn't exist download it and save it for the next time. |
| 74 | |
| 75 | Args: |
| 76 | root (Path or str, optional): The directory to load/save the data. If |
| 77 | none is given the ``~/.cache/mlx.data/mnist`` is used. |
| 78 | train (bool): Load the training or test set. |
| 79 | """ |
| 80 | |
| 81 | return _load_mnist_wrapper(root, train, "mnist") |
| 82 | |
| 83 | |
| 84 | def load_fashion_mnist(root=None, train=True): |
nothing calls this directly
no test coverage detected