benedekrozemberczki / SGCN

A PyTorch implementation of "Signed Graph Convolutional Network" (ICDM 2018).
GNU General Public License v3.0
268 stars 58 forks source link

fix classifier so that it will not always predict the majority class #7

Closed SherylHYX closed 3 years ago

SherylHYX commented 3 years ago

Fix classifier so that it will not always predict the majority class, also add one column to the output table, change README correspondingly as well.

There are some problems using a threshold determined by the ratio of negative links: On the one hand, you need to know the ground truth negative link ratio, which is a kind of cheating. On the other hand, due to the high imbalance of positive and negative links, your threshold would be close to zero, resulting in voting the majority class all the time.