Quickstart

Install the ECOLE dataset package.

pip3 install git+https://github.com/xingyaoww/ecole-dataset.git

For example, you can load all the evaluation datasets by running the following code:

import ecole_dataset
test_datasets = ecole_dataset.registry.load_datasets(
    split=ecole_dataset.SplitType.TEST
)

Please check scripts/example_usecase.ipynb for more examples.

NOTE: This package currently only supports loading dataset metadata (i.e., the test_datasets returned by .load_datasets does not contains pre-loaded images, but only image filepaths). We are actively working on allowing images to be also automatically loaded into test_datasets with a single line of code.

Please refer to docs/IMAGE_DATA.md to learn about how to download images and configure so that ecole-dataset can automatically loads images with one line of code.

Currently Supported Evaluation Datasets

Currently Supported Other Datasets