zfjsail / gae-pytorch

Graph Auto-Encoder in PyTorch
MIT License
417 stars 79 forks source link

ROC-AUC-score calculation function error #5

Closed Sowmya-R-Krishnan closed 4 years ago

Sowmya-R-Krishnan commented 4 years ago

In the code in utils.py, the function get_roc_score() has this line:

labels_all = np.hstack([np.ones(len(preds)), np.zeros(len(preds))])

The second argument should be np.zeros(len(preds_neg))

I was using your code to understand GVAE and found this small mistake. Thank you.

zfjsail commented 4 years ago

Thank you for pointing it out!