sunfanyunn / InfoGraph

Official code for "InfoGraph: Unsupervised and Semi-supervised Graph-Level Representation Learning via Mutual Information Maximization" (ICLR 2020, spotlight)
https://openreview.net/forum?id=r1lfF2NYvH
312 stars 45 forks source link

Linear Evaluation #2

Closed yyou1996 closed 4 years ago

yyou1996 commented 4 years ago

Hi,

Great job! I have a minor question. For unsupervised learning, how to choose the linear classifier? (LogReg, SVM ...)

sunfanyunn commented 4 years ago

We used LIBSVM as the downstream classifier.