ai4co / rl4co

A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)
https://rl4.co
MIT License
455 stars 84 forks source link

[Feature] Multi-agent environments with `min-max` objective #38

Closed fedebotu closed 1 year ago

fedebotu commented 1 year ago

As discussed yesterday with @alstn12088 , we could add also mTSP and mPDP, and include Equity-Transformer in the benchmark @Leaveson.

I have already made an mTSP implementation in free style here As for context and embeddings, you may find them here. Finally this notebook makes a simple training for mTSP - note that most probably you can add your knowledge and make it work better!

Extending to mPDP: I can make the environment, basically just extend mTSP to deal with coupled nodes (we already have implemented PDP here These multi-agent environments with min-max objective can be a fine addition to our collection :)

fedebotu commented 1 year ago

mPDP environment added! The Equity Transformer is work in progress as there may be some refactoring in terms of training with different agent numbers :)

fedebotu commented 1 year ago

mPDP environment added! The Equity Transformer is work in progress as there may be some refactoring in terms of training with different agent numbers :)