retrain: True
use_model_for_training: constant_abundance_p15_q05

overrides: {
  'tdust_grid_type': 'linear',
  'tdust_limits': [ 10, 35 ],
  'tdust_step': 2.5,
  'nh2_grid_type': 'log',
  'nh2_limits': [ 1e3, 6.5e7 ],
  'nh2_step': 1.316074013
}

#model_type: 'XGBoost'
#model_parameters: {
#  'n_estimators': 300,
#  'max_depth': 4,
#  'learning_rate': 0.02,
#  'subsample': 0.5
#}

use_validation: True

model_type: 'XGBoost_gridsearch'
model_parameters: {
  'param_grid':
    {
      'n_estimators': [ 300, 320, 50 ],
      'max_depth': [ 5, 7, 2 ],
      'learning_rate': [ 0.05, 0.1, 0.03 ],
      'subsample': [ 1, 2, 1 ],
      'lambda': [ 10, 20, 10 ],
      'colsample_bynode': [ 0.75, 0.8, 0.1 ],
#      'colsample_bylevel': [ 0.5, 1, 0.1 ],
      'colsample_bytree': [ 0.75, 0.8, 0.2 ],
      'min_child_weight': [ 100, 200, 100 ]
    },
  'param_gridsearch':
    {
      'cv': 5,
      'n_jobs': 16
    },
}
#model_parameters: {
#  'param_grid':
#    {
#      'n_estimators': [ 400, 900, 100 ],
#      'max_depth': [ 3, 7, 2 ],
#      'learning_rate': [ 0.01, 0.06, 0.02 ],
#      'subsample': [ 0.3, 0.9, 0.2 ]
#    },
#  'param_gridsearch':
#    {
#      'cv': 3,
#      'n_jobs': 16
#    },
#}

#model_type: 'auto_skl'
#model_parameters: {
#  'time_left_for_this_task': 3600,
#  'per_run_time_limit': 600,
#  'n_jobs': 12
#}

#model_type: 'RandomForest'
#model_parameters: {
#  'n_estimators': 20,
#  'criterion': 'friedman_mse',
#  'max_samples': 0.5
#}

