search

tkipf / gcn

Implementation of Graph Convolutional Networks in TensorFlow
MIT License
7.12k stars 2k forks source link

label problem #56

Open WowThankyou opened 6 years ago

WowThankyou commented 6 years ago

Hello everyone,

I have some problem about label, can the nodes in our own dataset have multi-label? In other word, the node could belong to two or more classes. If it can, could you please tell me how to implement ! Thank you very much!

tkipf commented 6 years ago

Yes, simply replace the loss function with a multi-label loss function (e.g. sigmoid cross entropy instead of softmax cross entropy).

WowThankyou commented 5 years ago

Thanks a lot!