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)