trainset graphs

[SPChytas]

Hello, and congratulations on your work. Is it possible to include the output of the first stage (the corresponding knowledge sub-graphs) for the train and val set too?

[cmavro]

Hi, thanks for the interest in our work!

Regarding training/validation sets, the ground-truth answers are known so we evaluate whether the GNN retrieves these answers. Thus, you can either use (i) the shortest paths between question and answer nodes, or (ii) the paths retrieved by GNN (you can do this during evaluation in this line for valid_data and train_data, setting write_info=True).

[SPChytas]

Thanks. I am interested in option (ii), i.e., use the paths that the GNN has retrieved. I tried to follow the instructions: "... or you can skip this folder and use directly the GNN output (retrieved answer nodes) that we computed (llm/results/gnn)." but I can't find the retrieved paths, beyond the test set.