search

AdrianArnaiz / DiffWire

DiffWire: Inductive Graph Rewiring via the Lovász Bound. In Proceedings of the First Learning on Graphs Conference. 2022. Adrian Arnaiz-Rodriguez, Ahmed Begga, Francisco Escolano and Nuria Oliver.
https://proceedings.mlr.press/v198/arnaiz-rodri-guez22a.html
16 stars 0 forks source link

Preprocess method #1

Open songminjae opened 1 year ago

songminjae commented 1 year ago

In train.py file, I guess there exists 3 main preprocessing methods. However, the only one working mehtod is digl. Are you willing to update the version of "sdrf" and "knn"?

AdrianArnaiz commented 1 year ago

Hi!

First, thanks a lot for reaching out and identifying this bug! The last time I used both sdrf and knn prepros, they were working smoothly. Let me run the code again to see if I have any issue with last modifications or versions and I'll get back to you asap :)

However, the main purpose of this repo is to provide the DiffWire code. Therefore, if you are in a hurry, SDRF prepro comes from the original repo and KNN comes from the original PyG version.