i = 4
print(f"""Example of graph snapshot {i}: \n
    Node feature shape: {xs[i].shape} \n
    Edge index shape: {edge_indices[i].shape} \n
    Edge feature shape: {edge_features[i].shape} \n 
    Labels shape: {ys[i].shape} \n
    Labels mask shape: {y_indices[i].shape}
    """)

Example of graph snapshot 4: 

    Node feature shape: (230, 2) 

    Edge index shape: (2, 703) 

    Edge feature shape: (703, 3) 
 
    Labels shape: (5,) 

    Labels mask shape: (2, 5)