Pretrained Cost Model for Distributed Constraint Optimization Problems
Requirements
- PyTorch 1.9.0
- PyTorch Geometric 1.7.1
Directory structure
baselinescontains the implementation of all compared baselinescorecontains the core data structures to run the simulationentrycontains the entry point of each algorithmheuristicscontains the implementation of GAT-PCM-boosted algorithmspretraincontains the implementation of pretraining stage
How to run the code
See the command line interface of run_*.py in entry.
Example:
python -um entry.run_dsa -pth problem.xml -c 1000 -p 0.8