diff --git a/automl_test.py b/automl_test.py index f3b574cd..c7645f99 100644 --- a/automl_test.py +++ b/automl_test.py @@ -39,8 +39,8 @@ model_in.set_index(index_columns, inplace=True) X_train, X_test, y_train, y_test = train_test_split(model_in.drop(["target", "pid"], axis=1), model_in["target"], test_size=0.20) automl = autosklearn.regression.AutoSklearnRegressor( - time_left_for_this_task=1200, - per_run_time_limit=60 + time_left_for_this_task=36000, + per_run_time_limit=120 ) automl.fit(X_train, y_train, dataset_name='straw') diff --git a/environment.yml b/environment.yml index cba49edc..536149bc 100644 --- a/environment.yml +++ b/environment.yml @@ -86,8 +86,6 @@ dependencies: - readline=8.0 - requests=2.25.0 - retrying=1.3.3 - - scikit-learn=0.23.2 - - scipy=1.5.2 - setuptools=51.0.0 - six=1.15.0 - smmap=3.0.4 @@ -107,34 +105,61 @@ dependencies: - zlib=1.2.11 - pip: - amply==0.1.4 + - auto-sklearn==0.14.7 - bidict==0.22.0 - biosppy==0.8.0 + - build==0.8.0 - cached-property==1.5.2 + - cloudpickle==2.2.0 - configargparse==0.15.1 + - configspace==0.4.21 - cr-features==0.2.1 - cycler==0.11.0 + - cython==0.29.32 + - dask==2022.2.0 - decorator==4.4.2 + - distributed==2022.2.0 + - distro==1.7.0 + - emcee==3.1.2 - fonttools==4.33.2 + - fsspec==2022.8.2 - h5py==3.6.0 + - heapdict==1.0.1 - hmmlearn==0.2.7 - ipython-genutils==0.2.0 - jupyter-core==4.6.3 - kiwisolver==1.4.2 + - liac-arff==2.5.0 + - locket==1.0.0 - matplotlib==3.5.1 + - msgpack==1.0.4 - nbformat==5.0.7 - opencv-python==4.5.5.64 - packaging==21.3 + - partd==1.3.0 - peakutils==1.3.3 + - pep517==0.13.0 - pillow==9.1.0 - pulp==2.4 + - pynisher==0.6.4 - pyparsing==2.4.7 + - pyrfr==0.8.3 - pyrsistent==0.15.5 - pywavelets==1.3.0 - ratelimiter==1.2.0.post0 + - scikit-learn==0.24.2 + - scipy==1.7.3 - seaborn==0.11.2 - shortuuid==1.0.8 + - smac==1.2 - snakemake==5.30.2 + - sortedcontainers==2.4.0 + - tblib==1.7.0 + - tomli==2.0.1 + - toolz==0.12.0 - toposort==1.5 + - tornado==6.2 - traitlets==4.3.3 - typing-extensions==4.2.0 + - zict==2.2.0 prefix: /opt/conda/envs/rapids