Data.DrCollate

Click here to view source code.

class DrCollate(follow_batch=None,
                exclude_keys=None):

It can be used to build the collate method in loading dataset.

PARAMETERS:

• follow_batch (optional) - Parameter of torch_geometric.data.Batch.from_data_list. (default: None)

• exclude_keys (optional) - Parameter of torch_geometric.data.Batch.from_data_list. (default: None)