Skip to content

Commit b44638e

Browse files
authored
change ensemble args logic
1 parent 8e9d19d commit b44638e

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

src/eschr/tl/main.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -177,7 +177,10 @@ def ensemble(
177177
hyperparam_iterator = [
178178
[k_range, la_res_range, metric, sparse, random_seed] for x in range(ensemble_size)
179179
]
180-
args = list(zip(data_iterator, hyperparam_iterator))
180+
sparse_iterator = repeat(sparse, ensemble_size)
181+
random_seed_iterator = repeat(random_seed, ensemble_size)
182+
process_id = list(range(ensemble_size))
183+
args = list(zip(data_iterator, hyperparam_iterator, sparse_iterator, random_seed_iterator, process_id))
181184

182185
print("Starting ensemble clustering multiprocess")
183186
out = parmap(run_base_clustering, args, nprocs=nprocs)

0 commit comments

Comments
 (0)