search

harryjo97 / GDSS

Official Code Repository for the paper "Score-based Generative Modeling of Graphs via the System of Stochastic Differential Equations" (ICML 2022)
https://arxiv.org/abs/2202.02514
139 stars 22 forks source link

File missing when sampling #2

Closed Frankie123421 closed 2 years ago

Frankie123421 commented 2 years ago

Hello, congratulations on your nice work and thanks for sharing the codes. When I sampled graphs with command "CUDA_VISIBLE_DEVICES=${gpu_ids} python main.py --type sample --config sample" as stated in README, I found that in sampler.py LINE 125: "with open(f'data/{self.configt.data.data.lower()}_test_nx.pkl', 'rb') as f:" the files "*_test_nx.pkl" are missing in data folder. I wonder how to fix it, thanks in advance.

harryjo97 commented 2 years ago

Hello, thanks for your interest in our work.

The files "*_test_nx.pkl" are created when you run the following command for the Preparations: python data/preprocess_for_nspdk.py --dataset ${dataset_name}

If you have any further questions, I would be happy to answer them!

Frankie123421 commented 2 years ago

Thanks, sorry for missing this command.