[Feat] Adding PolyNet model - Githubissues

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 PolyNet model #193

Closed ahottung closed 3 months ago

ahottung commented 3 months ago

Description

This PR adds the implementation of the PolyNet model (https://arxiv.org/abs/2402.14048).

Motivation and Context

PolyNet shows great performance on many problems and can easily be integrated in RL4CO.

Types of changes

[x] New feature (non-breaking change which adds core functionality)

Checklist

[x] I have updated the tests accordingly (required for a bug fix or a new feature).
[ ] I have updated the documentation accordingly.