config/automl_config.yaml

automl_library: autogluon

autogluon:
  time_limit: 600
  presets: medium_quality
  eval_metric: null
  verbosity: 2
  num_bag_folds: 5
  num_stack_levels: 1

flaml:
  time_budget: 600
  metric: auto
  task: classification
  estimator_list:
    - lgbm
    - xgboost
    - rf
    - extra_tree
  n_jobs: -1
  verbose: 1
  early_stop: true

pycaret:
  session_id: 42
  n_select: 5
  fold: 5
  verbose: false
  optimize: Accuracy
  use_gpu: false
  tuning:
    enabled: true
    n_iter: 10
    optimize: Accuracy
  ensemble:
    enabled: false
    method: Bagging
    n_estimators: 10

common:
  target_column: target
  problem_type: auto
  cv_folds: 5
  random_state: 42