JuliaGast / TGB2

Temporal Graph Benchmark project repo
1 stars 0 forks source link

Can we have dataset.num_nodes and dataset.num_rels #6

Closed JuliaGast closed 5 months ago

JuliaGast commented 6 months ago

It would be great to have these two properties for each dataset.

Otherwise I have to compute it for every method (also fine)

num_rels = len(set(dataset.edge_type))

num_nodes = len(set(dataset.full_data['sources']+ dataset.full_data['destinations']))

shenyangHuang commented 5 months ago

Yes, this can be supported in the next iteration.

shenyangHuang commented 5 months ago

issue addressed in #8

JuliaGast commented 5 months ago

great thanks. closing this.