diff options
| author | Lia Lenckowski <lialenck@protonmail.com> | 2024-03-03 19:40:16 +0100 |
|---|---|---|
| committer | Lia Lenckowski <lialenck@protonmail.com> | 2024-03-03 19:40:16 +0100 |
| commit | 3b670fe70de4b52b9a3f8614b42fb94fe49b3548 (patch) | |
| tree | 1119b13ab7119d8dc8bbaaebd77a9ec164eddac0 /src/main.rs | |
| parent | 94185d8b39af5a96a61ffb4df7a2d76dcd7afa49 (diff) | |
| download | embeddings-sort-3b670fe70de4b52b9a3f8614b42fb94fe49b3548.tar embeddings-sort-3b670fe70de4b52b9a3f8614b42fb94fe49b3548.tar.bz2 embeddings-sort-3b670fe70de4b52b9a3f8614b42fb94fe49b3548.tar.zst | |
add hash seed option for reproducible behaviour
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/main.rs b/src/main.rs index f9bf4fc..71e5d00 100644 --- a/src/main.rs +++ b/src/main.rs @@ -61,6 +61,10 @@ struct Args { #[arg(long, default_value = "christofides")] tsp_approx: TspAlg, + /// Seed for hashing. Random by default. + #[arg(long)] + hash_seed: Option<u64>, + images: Vec<PathBuf>, } @@ -138,7 +142,7 @@ where } let embeds: Vec<_> = embeds.into_iter().map(|e| e.unwrap()).collect(); - let (tsp_path, total_dist) = tsp(&embeds, &args.tsp_approx); + let (tsp_path, total_dist) = tsp(&embeds, &args.tsp_approx, &args.hash_seed); Ok(( tsp_path.iter().map(|i| args.images[*i].clone()).collect(), |