Skip to content

Commit

Permalink
change the default ensemble method to AlgoEnsembleBestByFold in AutoR… (
Browse files Browse the repository at this point in the history
#5722)

Change the default ensemble method

Signed-off-by: Mingxin Zheng
<18563433+mingxin-zheng@users.noreply.github.com>

Fixes Item 2 in #5564 .

### Description

A few sentences describing the changes proposed in this pull request.

### Types of changes
<!--- Put an `x` in all the boxes that apply, and remove the not
applicable items -->
- [x] Non-breaking change (fix or new feature that would not break
existing functionality).
- [x] Integration tests passed locally by running `./runtests.sh -f -u
--net --coverage`.

Signed-off-by: Mingxin Zheng <18563433+mingxin-zheng@users.noreply.github.com>
  • Loading branch information
mingxin-zheng committed Dec 13, 2022
1 parent 2b160b3 commit ff12159
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions monai/apps/auto3dseg/auto_runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -268,7 +268,7 @@ def __init__(

# intermediate variables
self.num_fold = 5
self.ensemble_method_name = "AlgoEnsembleBestN"
self.ensemble_method_name = "AlgoEnsembleBestByFold"
self.set_training_params()
self.set_prediction_params()
self.set_analyze_params()
Expand Down Expand Up @@ -473,7 +473,7 @@ def set_image_save_transform(self, kwargs):
output_dir=output_dir, output_postfix=output_postfix, output_dtype=output_dtype, resample=resample, **kwargs
)

def set_ensemble_method(self, ensemble_method_name: str = "AlgoEnsembleBestN", **kwargs):
def set_ensemble_method(self, ensemble_method_name: str = "AlgoEnsembleBestByFold", **kwargs):
"""
Set the bundle ensemble method
Expand Down

0 comments on commit ff12159

Please sign in to comment.