把最后的结果改成二分类

benedekrozemberczki / SimGNN

A PyTorch implementation of "SimGNN: A Neural Network Approach to Fast Graph Similarity Computation" (WSDM 2019).

GNU General Public License v3.0

759 stars 147 forks source link

Closed gr986 closed 3 years ago

gr986 commented 3 years ago

如果我想要把最后的结果改成二分类问题， 1、改动哪里的loss 2、改动哪里的输出使模型输出一个[batch_size,2]大小的的prediction，和target做交叉熵损失？