This package represents a community effort to provide a common interface for accessing common Machine Learning datasets. In contrast to other data-related Julia packages, the focus of MLDatasets.jl is specifically on downloading, unpacking, and accessing benchmark datasets. Functionality for the purpose of data processing or visualization is only provided to a degree that is special to some dataset.
This package is a part of the JuliaML ecosystem. Its functionality is built on top of the package DataDeps.jl.
Datasets are grouped into different categories. Click on the links below for a full list of datasets available in each category.
- Graphs - Datasets with an underlying graph structure: Cora, PubMed, CiteSeer, ...
- Misc - Datasets that do not fall into any of the other categories: Iris, BostonHousing, ...
- Text - Datasets for language models.
- Vision - Vision related datasets such as MNIST, CIFAR10, CIFAR100, ...
To install MLDatasets.jl, start up Julia and type the following code snippet into the REPL. It makes use of the native Julia package manger.
import Pkg
Pkg.add("MLDatasets")Pull requests contributing new datasets are warmly welcome. See the source code of any of the available implemented datasets for implementation examples.
If you don't find here the dataset you are looking for, please let us know by opening an issue. Moreover, you can check out these other packages to find what you need:
This code is free to use under the terms of the MIT license.