A benchmark of in-the-wild distribution shifts spanning diverse data modalities
WILDS
WILDS is a benchmark of in-the-wild distribution shifts spanning diverse data modalities and applications, from tumor identification to wildlife monitoring to poverty mapping.
Installation
We recommend using pip to install WILDS:
pip install wilds
If you have already installed it, please check that you have the latest version:
python -c "import wilds; print(wilds.__version__)"
# This should print "1.1.0". If it doesn't, update by running:
pip install -U wilds
If you plan to edit or contribute to WILDS, you should install from source:
git clone [email protected]:p-lambda/wilds.git
cd wilds
pip install -e .
Requirements
- numpy>=1.19.1
- ogb>=1.2.6
- outdated>=0.2.0
- pandas>=1.1.0
- pillow>=7.2.0
- pytz>=2020.4
- torch>=1.7.0
- torch-scatter>=2.0.5
- torch-geometric>=1.6.1
- tqdm>=4.53.0
Running pip install wilds or pip install -e . will automatically check for and