adding negative samples for polecat

JuliaGast / TGB2

Temporal Graph Benchmark project repo

1 stars 0 forks source link

Closed shenyangHuang closed 5 months ago

shenyangHuang commented 5 months ago

added tkg_negative_generator.py to generate negative samples for TKGs specifically based on time filtered MRR, currently very scalable, takes about 1 sec to generate all ns samples for validation set example
see polecat example for how to load the negative samples