ai4co / rl4co

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

[Feat] Adding support for improvement method #174

Closed yining043 closed 3 months ago

yining043 commented 4 months ago

Description

This PR is to make RL4CO support the improvement method for VRPs. The changes includes:

Motivation and Context

This PR is to make RL4CO support the improvement methods.

Types of changes

What types of changes does your code introduce? Remove all that do not apply:

Checklist

Go over all the following points, and put an x in all the boxes that apply. If you are unsure about any of these, don't hesitate to ask. We are here to help!

yining043 commented 4 months ago

Hi @cbhua @fedebotu , thank you so much for the review and great suggestions! I have replied above. Since last time I forgot to perform the pre-commit so I did a forced re-commit of the files. Sorry if it looks a bit hard to track the changes. I will perform new commits for future updates! :)

yining043 commented 4 months ago

Hi @LTluttmann, thanks for the review! I have changed the codes in the latest commit! I marked this pull request as draft since I need to add more features before merging~