Module: tff.simulation.datasets.inaturalist

Libraries for the federated iNaturalist dataset for simulation.

Classes

class ClientData: Object to hold a federated dataset.

class INaturalistSplit: The different split for the iNaturalist dataset.

Functions

get_synthetic(...): Returns a small synthetic dataset for testing.

load_data(...): Loads a federated version of the iNaturalist 2017 dataset.

INAT_TEST_DOWNLOAD_URL 'https://storage.googleapis.com/tff-datasets-public/iNaturalist_test.csv'
INAT_TEST_IMAGE_MD5_CHECKSUM '7d9b096fa1cd94d67a0fa779ea301234'
INAT_TEST_IMAGE_URL 'https://ml-inat-competition-datasets.s3.amazonaws.com/2017/test2017.tar.gz'
INAT_TEST_SPLIT_FILE 'iNaturalist_test.csv'
INAT_TEST_SPLIT_FILE_MD5_CHECKSUM 'bcf08a5a740ee76dac4e5af668081891'
INAT_TRAIN_DOWNLOAD_URL 'https://storage.googleapis.com/tff-datasets-public/iNaturalist_train.csv'
INAT_TRAIN_IMAGE_MD5_CHECKSUM '7c784ea5e424efaec655bd392f87301f'
INAT_TRAIN_IMAGE_URL 'https://ml-inat-competition-datasets.s3.amazonaws.com/2017/train_val_images.tar.gz'
INAT_TRAIN_SPLIT_FILE 'iNaturalist_train.csv'
INAT_TRAIN_SPLIT_FILE_MD5_CHECKSUM '63d5a41699434bc2b13f5853f3764c18'
LOGGER 'iNat2017'
TEST_FILE_NAME 'test.tfRecord'
TRAIN_SUB_DIR 'train'