Skip to content

Commit

Permalink
bugfix in greedy calling the multi order measures
Browse files Browse the repository at this point in the history
  • Loading branch information
Laouen authored Aug 16, 2024
1 parent 5da6a05 commit 2a85f7a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion thoi/heuristics/greedy.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@ def greedy(X:np.ndarray,
assert metric in ['tc', 'dtc', 'o', 's'], f'metric must be one of tc, dtc, o or s. invalid value: {metric}'

current_solution = multi_order_measures(
X, initial_order, initial_order, batch_size=batch_size, use_cpu=use_cpu,
X, min_order=initial_order, max_order=initial_order, batch_size=batch_size, use_cpu=use_cpu,
batch_data_collector=partial(batch_to_tensor, top_k=repeat, metric=metric, largest=largest),
batch_aggregation=partial(concat_tensors, top_k=repeat, metric=metric, largest=largest)
)[-1]
Expand Down

0 comments on commit 2a85f7a

Please sign in to comment.