CombOptNet

CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints

This repository contains PyTorch implementation of the paper CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints

Installation

  1. Run pipenv install (at your own risk with --skip-lock to save some time).
  2. From within the pipenv environment run python3 -m pip install -i https://pypi.gurobi.com gurobipy.
  3. Obtain a license and download/set it.
  4. Download and extract the datasets.

Usage

For [experiment] = knapsack or [experiment] = static_constraints:

  1. Set the base_dataset_path parameter in experiments/[experiment]/base.yaml.
  2. In case of static constraints: set the dataset_specification parameter in experiments/static_constraints/base.yaml
  3. Run python3 main.py experiments/[experiment]/[method].yaml.

Citation

@misc{paulus2021comboptnet,
      title={CombOptNet: Fit the Right NP-Hard Problem by Learning Integer Programming Constraints}, 
      author={Anselm Paulus and Michal Rolínek and Vít Musil and Brandon Amos and Georg Martius},
      year={2021},
      eprint={2105.02343},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

GitHub

https://github.com/martius-lab/CombOptNet