Skip to content

Commit 6d12d9c

Browse files
committed
reproduction instructions in README
1 parent b5368c3 commit 6d12d9c

1 file changed

Lines changed: 19 additions & 0 deletions

File tree

README.md

Lines changed: 19 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@ License|Linux, MacOS & Windows Build|Code Coverage|Zenodo
77
Table of Contents
88
-----------
99

10+
* [SEA 2026 Reproduction](#sea-2026-sea)
1011
* [About Mt-KaHyPar](#about-mt-kahypar)
1112
* [Features](#features)
1213
* [Installing Mt-KaHyPar](#installing-mt-kahypar)
@@ -18,6 +19,24 @@ Table of Contents
1819
* [Improving Compile Times](#improving-compile-times)
1920
* [Licensing](#licensing)
2021

22+
SEA 2026 Reproduction
23+
-----------
24+
This is a short guide how to reproduce the results from our SEA 2026 paper "Engineering Learned Heuristics to Improve Clustering for Multilevel Graph Partitioning".
25+
The code is configured such that the default preset executes our ML guided coarsening algorithm from the paper.
26+
I.e., running Mt-KaHyPar (see [below](#running-mt-kahypar)) with the following parameters will reproduce the results of our main configuration:
27+
```
28+
--preset=default --input-file-format=metis
29+
```
30+
31+
To instead reproduce the results of our modified baseline, you need to also add the parameters:
32+
```
33+
--c-guiding-by-integrated-model=false --c-rating-degree-similarity-policy=always_accept
34+
```
35+
36+
To reproduce results for different ML models than our final model, a small modification of the source code is necessary.
37+
The header file `mt-kahypar/partition/coarsening/multilevel/ml/model_definition.h` defines which of the ML models is used.
38+
Swapping out the include statement and then recompiling Mt-KaHyPar will allow to use a different model.
39+
2140
About Mt-KaHyPar
2241
-----------
2342
Mt-KaHyPar is a shared-memory algorithm for partitioning graphs and hypergraphs. The balanced (hyper)graph partitioning problem

0 commit comments

Comments
 (0)