Giters

Samyu0304 / graph-information-bottleneck-for-Subgraph-Recognition

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

gnn-modelinformation-theory-and-codeiclr2021interpretable-machine-learning

graph-information-bottleneck-for-subgraph-recognition

Code for the paper "Graph Information Bottleneck for Subgraph Recognition" in ICLR 2021. Something important: there is a MI_est function where I implement multiple estimators in trian_eval.py of graph classfication task. Please use Donsker-varadhan estimator as recommended in our paper instead of the last one.

About

gnn-modelinformation-theory-and-codeiclr2021interpretable-machine-learning

License:MIT License

Languages

Language:Python 99.7%Language:Shell 0.3%