tomsilver / ploi

Code for "Planning with Learned Object Importance in Large Problem Instances using Graph Neural Networks" (AAAI 2021)
MIT License
14 stars 9 forks source link

PLOI

This repository houses code for the AAAI 2021 paper:

Planning with Learned Object Importance in Large Problem Instances using Graph Neural Networks

Tom Silver, Rohan Chitnis, Aidan Curtis, Joshua Tenenbaum, Tomas Lozano-Perez, Leslie Pack Kaelbling.

For any questions or issues with the code, please email ronuchit@gmail.com and tslvr@mit.edu.

Link to paper: https://arxiv.org/abs/2009.05613

Link to video: https://www.youtube.com/watch?v=FWsVJc2fvCE

Instructions for running (tested on Mac and Linux):

Now, ./run.sh should work. Different domains and methods can be run by modifying the variables at the top of run.sh.