Zerg-Overmind/QC-DGM

Questions about evaluation details

xmlyqing00 opened this issue · 0 comments

Hi,

Thank you for providing the training and evaluation codes. In the evaluation part, you solve the quadratic constraint (Eqn 7) to get the optimal assignment. In your code, the QC-optimization starts from line 128 in QCDGM/model.py. The quadratic constraints are AA_src and BB_tgt

After that, I think the assignment is stored in the variable s (or X? I'm confused here, I think it should be X but you return s instead) and it is returned to eval.py as s_pred.

You do a Hungarian algorithm here (eval.py line 81) to discrete the values. That makes sense. But I can't follow the following several lines. You do the qc_opt again with constraints A_src and A_tgt, which represents edge connections. Why do you need to do it as "post-processing"? In your training codes, I don't see such post-processing there.

Thanks for your time.