Closed JuliaGast closed 5 months ago
for example: for tkg_negative_generator.py
add the arginverse_triples_flag (default: True)
if inverse_triples_flag
add negatives for both directions
Given that for all tkg
datasets, we need to add inverse relations, I think we should bake this directly into the dataloader. This means that the edges retrieved from dataloader will contain already the inverted relations and for the negative sampler, it would just function as normal (to generate for both directions)
This has been implemented in PR #14