multiLabelGCN

Code with GCN

https://github.com/dragen1860/GCN-PyTorch

소개

Baseline의 GCN은 multi-label일 때 적합하지 않다.

이에, torch.nn.MultiLabelSoftMarginLoss를 이용하여 multi-label일 때의 GCN을 구현해보았다.