DeepGraphLearning / RNNLogic

120 stars 25 forks source link

How does generator update? #10

Open Jessielym66 opened 2 years ago

Jessielym66 commented 2 years ago

I'm confused about the EM step in codes. The paper describes that the rule generator is updated by maximizing the log-likelyhood of the identified high-quality rules, while in the codes, rule generator computes loss between 'logits' and 'target'. I wonder that whether the 'logits' is the first term 'RNN(rule|r)' in Eq.(9)? And what does the 'target' mean? Why do you apply CrossEntropyLoss on them? Does it achieve the efficiency same as maximization? Or do I misunderstand the code?