def test_qm7(): dataset = datasets.QM7() dl = DisjointLoader(dataset, batch_size=batch_size) dl.__next__() bl = BatchLoader(dataset, batch_size=batch_size) bl.__next__()
def test_qm9(): dataset = datasets.QM9(amount=100) dl = DisjointLoader(dataset, batch_size=batch_size) dl.__next__() bl = BatchLoader(dataset, batch_size=batch_size) bl.__next__()
def test_tud(): # Edge labels + edge attributes dataset = datasets.TUDataset('BZR_MD', clean=False) dl = DisjointLoader(dataset, batch_size=batch_size) dl.__next__() bl = BatchLoader(dataset, batch_size=batch_size) bl.__next__() # Node labels + node attributes + clean version dataset = datasets.TUDataset('ENZYMES', clean=True) dl = DisjointLoader(dataset, batch_size=batch_size) dl.__next__() bl = BatchLoader(dataset, batch_size=batch_size) bl.__next__()