search

SateeshKumar21 / graph-inverse-rl

MIT License
32 stars 4 forks source link

Spatial Interaction Encoder #4

Open AnnabellaMacaluso opened 3 months ago

AnnabellaMacaluso commented 3 months ago

Hi! I saw there's a script called compute_goal_embedding.py I was wondering if this is included in the repository? Is the Spatial Interaction Encoder Network code included in this repository? Thanks!

SateeshKumar21 commented 3 months ago

Hi, thanks for raising the issue.

I realized I had missed some important files. I have added them in, please refer to compute_goal_embedding.py and models.py for your questions. There might still be some inconsistencies in the codebase since I have just put the files here and haven't tried to run it yet. I intend to resolve those later on, for now I hope the code serves as a good reference point for starting out with GraphIRL.

AnnabellaMacaluso commented 1 month ago

Hi! Thanks for adding the code. Correct me if I'm wrong but it seems the Spatial Interaction Network is trained first? I was wondering where I can find the training code for this, I see lots of Trainers but I don't see any training script.