imports partially not aligned with file structure

[hagreven]

Hi, thanks for sharing the code. I want to run some experiments with it for my bachelor thesis.
Trying to run the jupyter notebook kindly provided, I found a few issues that also hold for running the code from main.py.
The first category of issues I encountered concerns the import statements in 4 files.

[danyang-liu]

Thanks, I will fix it soon

[ZJJHYM]

Do you know how to download dataset "data_mind_small.pkl"?Thanks!

[hagreven]

Sorry, no. Maybe have a look at the provided jupyter notebook. That's what I used for downloading the datasets.