ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

Home Page:https://rl4.co

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[Feature] various encoder implement

cbhua opened this issue · comments

Implement different encoders to show that our repository implementation is flexible, and it could add more impact rather than going for the MatNet.

Tasks:

  • MPNN encoder;
  • GCN encoder;

Other encoders for future work.

Perfect!
I will be fixing the last things to start training on Monday in the meantime

Done; after having a test, we can close this issue.