Skip to content

Commit

Permalink
Fix #91: Enable Dynamic Stacking as default
Browse files Browse the repository at this point in the history
  • Loading branch information
AnirudhDagar authored Nov 7, 2024
1 parent 932f931 commit 813f5f7
Showing 1 changed file with 0 additions and 9 deletions.
9 changes: 0 additions & 9 deletions src/autogluon_assistant/predictor.py
Original file line number Diff line number Diff line change
Expand Up @@ -83,15 +83,6 @@ def fit(self, task: TabularPredictionTask) -> "AutogluonTabularPredictor":
logger.info(f"predictor_init_kwargs: {predictor_init_kwargs}")
logger.info(f"predictor_fit_kwargs: {predictor_fit_kwargs}")

if predictor_fit_kwargs.get("dynamic_stacking", False):
# Use config value for num_stack_levels
# Default 1 if dynamic_stacking is True
if predictor_fit_kwargs["num_stack_levels"] == 0:
predictor_fit_kwargs["num_stack_levels"] = 1
logger.info(
f"Dynamic stacking is enabled; setting num_stack_levels={predictor_fit_kwargs['num_stack_levels']}"
)

self.metadata |= {
"predictor_init_kwargs": predictor_init_kwargs,
"predictor_fit_kwargs": predictor_fit_kwargs,
Expand Down

0 comments on commit 813f5f7

Please sign in to comment.