search

Shen-Lab / GraphCL_Automated

[ICML 2021] "Graph Contrastive Learning Automated" by Yuning You, Tianlong Chen, Yang Shen, Zhangyang Wang; [WSDM 2022] "Bringing Your Own View: Graph Contrastive Learning without Prefabricated Data Augmentations" by Yuning You, Tianlong Chen, Zhangyang Wang, Yang Shen
108 stars 11 forks source link

Unsupervised_TU - KeyError :numnodes #41 #5

Open leesoohong opened 2 years ago

leesoohong commented 2 years ago

When i run go.sh with dataset except "NCI1", "PROTEINS", "DD", "MUTAG" i got log like this. Please let me know how to figure out this problem thank you image

yyou1996 commented 2 years ago

Hi @leesoohong,

Really sry for the late reply! I think this might result from the library version issue that when you call dataset.get_num_feature(), it does not give you num_nodes key. To fix it, I suppose you just modify this line with dataset[0].get_num_feature() (plz print it out and double-check).