Awesome
mgap
The code for "Masked Graph Auto-Encoder Constrained Graph Pooling" (ECML-PKDD 2022)
Requirements
conda create -c conda-forge -n my-rdkit-env rdkit python=3.7
pip install torch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0
pip install torch-scatter torch-sparse torch-cluster torch-spline-conv torch-geometric -f https://pytorch-geometric.com/whl/torch-1.9.0+cu102.html
pip install transformers
Run Code
python main.py --model SAG --dataset DD --loss_alpha 0 --loss_beta 0