updated default config. (#91)

autogluon · Nov 6, 2024 · 932f931 · 932f931
1 parent 8020e1a
commit 932f931
Show file tree

Hide file tree

Showing 3 changed files with 0 additions and 12 deletions.
diff --git a/src/autogluon_assistant/configs/best_quality.yaml b/src/autogluon_assistant/configs/best_quality.yaml
@@ -22,12 +22,8 @@ feature_transformers:
 autogluon:
   predictor_init_kwargs: {}
   predictor_fit_kwargs:
-    verbosity: 2
     presets: best_quality
     time_limit: 14400
-    dynamic_stacking: True
-    num_bag_folds: 5
-    num_stack_levels: 2
 llm:
   # Note: bedrock is only supported in limited AWS regions
   provider: bedrock

diff --git a/src/autogluon_assistant/configs/high_quality.yaml b/src/autogluon_assistant/configs/high_quality.yaml
@@ -9,12 +9,8 @@ feature_transformers:
 autogluon:
   predictor_init_kwargs: {}
   predictor_fit_kwargs:
-    verbosity: 2
     presets: high_quality
     time_limit: 3600
-    dynamic_stacking: True
-    num_bag_folds: 5
-    num_stack_levels: 2
 llm:
   # Note: bedrock is only supported in limited AWS regions
   provider: bedrock

diff --git a/src/autogluon_assistant/configs/medium_quality.yaml b/src/autogluon_assistant/configs/medium_quality.yaml
@@ -9,12 +9,8 @@ feature_transformers:
 autogluon:
   predictor_init_kwargs: {}
   predictor_fit_kwargs:
-    verbosity: 2
     presets: medium_quality
     time_limit: 600
-    dynamic_stacking: True
-    num_bag_folds: 5
-    num_stack_levels: 2
 llm:
   # Note: bedrock is only supported in limited AWS regions
   provider: bedrock