From 0c0ffc69529b4058b36d50e5c59b8d8eb1bfc8ae Mon Sep 17 00:00:00 2001 From: jackjii79 Date: Fri, 20 Sep 2024 19:05:20 -0700 Subject: [PATCH] upgrade autogluon --- models/algorithms/autogluon.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/algorithms/autogluon.py b/models/algorithms/autogluon.py index 58bab8c0..3d24a0fe 100644 --- a/models/algorithms/autogluon.py +++ b/models/algorithms/autogluon.py @@ -29,7 +29,7 @@ class AutoGluonModel(CustomModel): # so use isolated env isolate_env = dict(pyversion="3.11", install_h2oaicore=False, install_datatable=True, cache_env=True, cache_by_full_module_name=False, install_pip="latest", - modules_needed_by_name=['autogluon==0.3.1']) + modules_needed_by_name=['autogluon==1.0.0']) @staticmethod def is_enabled(): @@ -115,7 +115,7 @@ def fit_static(X, y, sample_weight=None, eval_set=None, sample_weight_eval_set=N 'XGB': {'n_jobs': n_jobs, 'ag_args_fit': {'num_gpus': num_gpus, 'num_cpus': n_jobs}}, 'CAT': {'thread_count': n_jobs, 'ag_args_fit': {'num_gpus': num_gpus, 'num_cpus': n_jobs}}, 'GBM': [{}, {'extra_trees': True, 'ag_args': {'name_suffix': 'XT'}}, 'GBMLarge'], - 'NN': {'ag_args_fit': {'num_gpus': num_gpus, 'num_cpus': n_jobs}}, + 'NN_TORCH': {'ag_args_fit': {'num_gpus': num_gpus, 'num_cpus': n_jobs}}, 'FASTAI': {'ag_args_fit': {'num_gpus': num_gpus, 'num_cpus': n_jobs}}, } kwargs_fit = dict(hyperparameters=hyperparameters)