tkipf / c-swm

Contrastive Learning of Structured World Models
MIT License
386 stars 67 forks source link

Any scripts for representation visualization? #5

Open NagisaZj opened 3 years ago

NagisaZj commented 3 years ago

Hello, thank you for sharing the codes. I really like the Figures 3 and 4 in your paper, and wonder if there is any scripts for reproducing these results. Thanks!

tkipf commented 3 years ago

Thanks a lot for the kind words! I found two ipynb files that I still had lying around that contain the visualization scripts. It's a bit messy and some of the function names in the released code have changed, but maybe it's helpful: https://www.dropbox.com/s/v269hjji8e9rx6g/c_swm_visualization.zip?dl=0