Datasets

The datasets module provides loaders for small bundled datasets and fetchers for larger downloadable datasets.

Function	Dataset
`load_horsager2009()`	Threshold data from [Horsager2009]
`load_nanduri2012()`	Brightness ratings from [Nanduri2012]
`load_perezfornos2012()`	Phosphene fading from [PerezFornos2012]
`fetch_beyeler2019()`	Phosphene drawings from [Beyeler2019]
`fetch_han2021()`	Outdoor scenes from [Han2021]

Pandas is required for tabular datasets; some datasets also require HDF5 via h5py.

Downloaded data

Fetched datasets are cached in ~/pulse2percept_data by default. Set the PULSE2PERCEPT_DATA environment variable or pass a data directory directly to a fetcher to use another location.

import pulse2percept as p2p

p2p.datasets.get_data_dir()
p2p.datasets.clear_data_dir()

clear_data_dir removes the cached files, so the next fetch downloads them again.