data4co 0.0.1a2

data4co provides convenient dataset generators for the combinatorial optimization problem

Data4CO

A data generator tool for Combinatorial Optimization (CO) problems, enabling customizable, diverse, and scalable datasets for benchmarking optimization algorithms.

Current support

version: 0.0.1-alpha

Problem	Solver1	Impl.	Solver2	Impl.	Solver3	Impl.
TSP	LKH	✔	Concorde	✔	Gurobi	📆
MIS	KaMIS	✔	Gurobi	📆	--	--

Problem	Type1	Impl.	Type2	Impl.	Type3	Impl.	Type4	Impl.
TSP	uniform	✔	gaussian	✔	cluster	📆	--	--
MIS	ER	✔	BA	✔	HK	✔	WS	✔

✔: Supported; 📆: Planned for future versions (contributions welcomed!).

How to Install

Github

Clone with the url https://github.com/heatingma/Data4CO.git , and the following packages are required, and shall be automatically installed by pip:

Python >= 3.8
numpy>=1.24.4
networkx==2.8.8
lkh>=1.1.1
tsplib95==0.7.1
tqdm>=4.66.1

PyPI

It is very convenient to directly use the following commands

pip install data4co

How to Use

TSP

from data4co import TSPDataGenerator

tsp_data_lkh = TSPDataGenerator(
    batch_size=16,
    nodes_num=50,
    data_type="uniform",
    solver_type="lkh",
    train_samples_num=128000,
    val_samples_num=1280,
    test_samples_num=1280,
    save_path="your/path/to/save"
)

tsp_data_lkh.generate()

MIS

from data4co import MISDataGenerator

mis_data_kamis = MISDataGenerator(
    nodes_num_min=700, 
    nodes_num_max=800,
    data_type="er",
    solver_type="kamis",
    train_samples_num=128000,
    val_samples_num=1280,
    test_samples_num=1280,
    save_path="your/path/to/save",
    solve_limit_time=10.0
)

mis_data_kamis.generate()
mis_data_kamis.solve()