Why building test graph requires train_triplets?

[zhaohan-xi]

Hi Dear Author, in this line, building the test_graph has train triplets as input, isn't it test triplets actually?

[JinheonBaek]

Thank you for your question.

This is a nice question, and we use the test_graph (all train triplets) to get embeddings of entities on the given graph, which are further used to test the model on the test triplets. In other words, the test triplet's entity embedding is obtained from the RGCN outputs of the test_graph (train_triplets).