Query generation only happen with test sets in QueryGenerator class

[sshivam95]

In the method generate_queries, the mode should decide which edges should be passed to ground queries for getting queries and answers (as per the docstring). However, only test set is used. There is no functionality to use just the train or validation set.

[Demirrr]

the mode should decide which edges should be passed to ground queries for getting queries and answers (as per the docstring).

What do you mean by the mode ?

However, only test set is used. There is no functionality to use just the train or validation set.

Since A KGE model is not trained on the query answer set, there is no train dataset is generated. Do you think that creating queries over train or validation set would be usefull ?