[FEATURE REQUEST] provide more information in dataloaders for graph

[Jn-Huang]

Is your feature request related to a problem? Please describe.

When we load a set of graphs, we have no information on the difference between datasets. For example,

where we only get three datasets.

Describe the solution you'd like

Indicate that the three datasets are train_dataset, val_dataset, test_dataset respectively.