Skip to content

Tfds

dataphy.sources.tfds

Functions

fetch(source: Dict, output_dir: str, include: str, runner: str = 'DirectRunner')

Source code in src/dataphy/sources/tfds.py 
 6
 7
 8
 9
10
11
12
13
def fetch(source: Dict, output_dir: str, include: str, runner: str = "DirectRunner"):
    import tensorflow_datasets as tfds

    name = source.get("name")
    split = source.get("split", "train")
    data_dir = source.get("data_dir") or output_dir
    tfds.load(name, split=split, data_dir=data_dir, download=True)
    return str(pathlib.Path(data_dir))