Changes in AutoML params and environment.yml

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')
 

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