def LoadData(DATASET_NAME):
"""
This function is called in the main.py file
returns:
; dataset object
"""
# handling for MNIST or CIFAR Superpixels
if DATASET_NAME == 'MNIST' or DATASET_NAME == 'CIFAR10':
return SuperPixDataset(DATASET_NAME)
# handling for (ZINC) experiments dataset
if DATASET_NAME == 'ZINC':
return MoleculeDataset(DATASET_NAME)
# handling for the TU Datasets
TU_DATASETS = ['COLLAB', 'ENZYMES', 'DD', 'PROTEINS_full']
if DATASET_NAME in TU_DATASETS:
return TUsDataset(DATASET_NAME)
# handling for SBM datasets
SBM_DATASETS = ['SBM_CLUSTER', 'SBM_PATTERN']
if DATASET_NAME in SBM_DATASETS:
return SBMsDataset(DATASET_NAME)
# handling for TSP dataset
if DATASET_NAME == 'TSP':
return TSPDataset(DATASET_NAME)
# handling for the CITATIONGRAPHS Datasets
CITATIONGRAPHS_DATASETS = ['CORA', 'CITESEER', 'PUBMED']
if DATASET_NAME in CITATIONGRAPHS_DATASETS:
return CitationGraphsDataset(DATASET_NAME)
def LoadData(DATASET_NAME):
"""
This function is called in the main.py file
returns:
; dataset object
"""
# handling for MNIST or CIFAR Superpixels
if DATASET_NAME == 'MNIST' or DATASET_NAME == 'CIFAR10':
return SuperPixDataset(DATASET_NAME)
# handling for (ZINC) molecule dataset
if DATASET_NAME == 'ZINC':
return MoleculeDataset(DATASET_NAME)
# handling for the TU Datasets
TU_DATASETS = [
'BZR', 'COX2', 'DHFR', 'ENZYMES', 'PROTEINS_full', 'Synthie'
]
if DATASET_NAME in TU_DATASETS:
return TUsDataset(DATASET_NAME)
# handling for SBM datasets
SBM_DATASETS = ['SBM_CLUSTER', 'SBM_PATTERN']
if DATASET_NAME in SBM_DATASETS:
return SBMsDataset(DATASET_NAME)
# handling for TSP dataset
if DATASET_NAME == 'TSP':
return TSPDataset(DATASET_NAME)
def LoadData(DATASET_NAME):
"""
This function is called in the main.py file
returns:
; dataset object
"""
# handling for MNIST or CIFAR Superpixels
if DATASET_NAME == 'MNIST' or DATASET_NAME == 'CIFAR10':
return SuperPixDataset(DATASET_NAME)
# handling for (ZINC) molecule dataset
if DATASET_NAME == 'ZINC' or DATASET_NAME == 'ZINC-full':
return MoleculeDataset(DATASET_NAME)
# handling for the TU Datasets
TU_DATASETS = ['ENZYMES', 'DD', 'PROTEINS_full']
if DATASET_NAME in TU_DATASETS:
return TUsDataset(DATASET_NAME)
# handling for SBM datasets
SBM_DATASETS = ['SBM_CLUSTER', 'SBM_PATTERN']
if DATASET_NAME in SBM_DATASETS:
return SBMsDataset(DATASET_NAME)
# handling for TSP dataset
if DATASET_NAME == 'TSP':
return TSPDataset(DATASET_NAME)
# handling for COLLAB dataset
if DATASET_NAME == 'OGBL-COLLAB':
return COLLABDataset(DATASET_NAME)
# handling for the CSL (Circular Skip Links) Dataset
if DATASET_NAME == 'CSL':
return CSLDataset(DATASET_NAME)