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