def load_mutag(): try: dataset = fetch_dataset("MUTAG", with_classes=True, verbose=verbose) except Exception: # Offline testing warn('MUTAG could not be downloaded: using an offline version..') cwd = os.getcwd() os.chdir(os.path.join(os.path.dirname(__file__), 'data')) dataset = read_data('MUTAG', with_classes=True) os.chdir(cwd) return dataset.data, dataset.target
# The baseline dataset for node, edge_labels global dataset, dataset_tr, dataset_te try: dataset = fetch_dataset(dataset_name, with_classes=False, verbose=verbose).data except Exception: # Offline testing warn('There was a problem fetching dataset for attributes: [' + dataset_name + ']') if dataset_name != 'MUTAG': warn('Switching back to baseline dataset MUTAG') warn('Using an offline version..') cwd = os.getcwd() os.chdir(os.path.join(fdir, 'data')) dataset = read_data('MUTAG', with_classes=False).data os.chdir(cwd) dataset_tr, dataset_te = train_test_split(dataset, test_size=0.2, random_state=42) # The baseline dataset for node/edge-attributes global dataset_attr, dataset_attr_tr, dataset_attr_te try: dataset_attr = fetch_dataset(dataset_attr_name, with_classes=False, prefer_attr_nodes=True, prefer_attr_edges=True, verbose=verbose).data