Where is the atom loss and bond loss?

zaixizhang / MGSSL

Official implementation of NeurIPS'21 paper"Motif-based Graph Self-Supervised Learning for Molecular Property Prediction"

Apache License 2.0

107 stars 15 forks source link

Where is the atom loss and bond loss? #8

Closed windyLemon closed 1 year ago

windyLemon commented 1 year ago

Dear author, in the paper the final loss is the sum of atom, bond and motif. I cannot find the loss of atom and bond in the source code. So what is the final loss function of MGSSL?

zaixizhang commented 1 year ago

For the conciseness of the code, we here only upload the motif-based SSL task codes in the GitHub repo. The atom/bond loss code can be found in related works such as https://github.com/snap-stanford/pretrain-gnns/