gds package

Subpackages

• gds.common package
  • Subpackages
    • gds.common.metrics package
      • Submodules
      • gds.common.metrics.all_metrics module
      • gds.common.metrics.loss module
      • gds.common.metrics.metric module
      • Module contents
  • Submodules
  • gds.common.data_loaders module
  • gds.common.grouper module
  • gds.common.utils module
  • Module contents
• gds.datasets package
  • Submodules
  • gds.datasets.colored_mnist_dataset module
  • gds.datasets.download_utils module
  • gds.datasets.gds_dataset module
  • gds.datasets.ogbgppa_dataset module
  • gds.datasets.ogbmolhiv_dataset module
  • gds.datasets.ogbmolpcba_dataset module
  • gds.datasets.pyg_colored_mnist_dataset module
  • gds.datasets.pyg_rotated_mnist_dataset module
  • gds.datasets.pyg_sbm_environment_dataset module
  • gds.datasets.pyg_sbm_isolation_dataset module
  • gds.datasets.pyg_upfd_dataset module
  • gds.datasets.rotated_mnist_dataset module
  • gds.datasets.sbm_environment_dataset module
  • gds.datasets.sbm_isolation_dataset module
  • gds.datasets.upfd_dataset module
  • Module contents

Submodules

gds.download_datasets module

gds.download_datasets.main()

Downloads the latest versions of all specified datasets, if they do not already exist.

gds.get_dataset module

gds.get_dataset.get_dataset(dataset, version=None, **dataset_kwargs)

Returns the appropriate WILDS dataset class. Input:

dataset (str): Name of the dataset version (str): Dataset version number, e.g., ‘1.0’.

Defaults to the latest version.

dataset_kwargs: Other keyword arguments to pass to the dataset constructors.

Output:

The specified WILDSDataset class.

gds.version module

gds.version.check()

Module contents