search

Shen-Lab / GraphCL

[NeurIPS 2020] "Graph Contrastive Learning with Augmentations" by Yuning You, Tianlong Chen, Yongduo Sui, Ting Chen, Zhangyang Wang, Yang Shen
MIT License
548 stars 103 forks source link

Question about cortex_DIM in net_infomax.py of semisupervised_TU/finetuning #68

Open ZhengXR-L opened 8 months ago

ZhengXR-L commented 8 months ago

There is a package import _'from cortex_DIM.functions.gan_losses import get_positive_expectation, get_negativeexpectation' Neither '_pip install cortexDIM' nor '_conda install cortexDIM' could work.

I wander how can i add this cortex_DIM to my environment? Or anyone can tell me what _get_positive_expectation, get_negativeexpectation are doing? I need them in _lossinfomax in net_infomax.py

yyou1996 commented 8 months ago

Hi,

I think for the cortex_DIM package you can refer to the https://github.com/sunfanyunn/InfoGraph package, which we built on for this experiment (and we didn't use it, cortex_DIM).

ZhengXR-L commented 8 months ago

Thanks for your reply.

The codes work well after I add necessary packages provided by you.