MAINT make each example run 120 seconds

Author: mfeurer

autosklearn/automl.py

@@ -376,11 +376,20 @@ def _fit(self, datamanager):
                                  "left.")
             self._proc_smac = None
         else:
+            if self._per_run_time_limit is None or \
+                    self._per_run_time_limit > time_left_for_smac:
+                print('Time limit for a single run is higher than total time '
+                      'limit. Capping the limit for a single run to the total '
+                      'time given to SMAC (%f)' % time_left_for_smac)
+                per_run_time_limit = time_left_for_smac
+            else:
+                per_run_time_limit = self._per_run_time_limit
+
             self._proc_smac = AutoMLSMBO(config_space=self.configuration_space,
                                          dataset_name=self._dataset_name,
                                          backend=self._backend,
                                          total_walltime_limit=time_left_for_smac,
-                                         func_eval_time_limit=self._per_run_time_limit,
+                                         func_eval_time_limit=per_run_time_limit,
                                          memory_limit=self._ml_memory_limit,
                                          data_memory_limit=self._data_memory_limit,
                                          watcher=self._stopwatch,

example/example_crossvalidation.py

@@ -18,7 +18,7 @@ def main():
     X_test = X[1000:]
     y_test = y[1000:]
     automl = autosklearn.classification.AutoSklearnClassifier(
-        time_left_for_this_task=60, per_run_time_limit=30,
+        time_left_for_this_task=120, per_run_time_limit=30,
         tmp_folder='/tmp/autoslearn_cv_example_tmp',
         output_folder='/tmp/autosklearn_cv_example_out',
         delete_tmp_folder_after_terminate=False,

example/example_parallel.py

@@ -35,7 +35,8 @@ def spawn_classifier(seed, dataset_name):
     # models.
     # 3. all instances of the AutoSklearnClassifier must have a different seed!
     automl = AutoSklearnClassifier(
-        time_left_for_this_task=180, # sec., how long should this seed fit process run
+        time_left_for_this_task=120, # sec., how long should this seed fit
+        # process run
         per_run_time_limit=60, # sec., each model may only take this long before it's killed
         ml_memory_limit=1024, # MB, memory limit imposed on each call to a ML  algorithm
         shared_mode=True, # tmp folder will be shared between seeds

example/example_regression.py

@@ -14,7 +14,7 @@ def main():
     np.random.shuffle(indices)
     X_train, X_test, y_train, y_test = train_test_split(X, y)
     automl = autosklearn.regression.AutoSklearnRegressor(
-        time_left_for_this_task=600,
+        time_left_for_this_task=120, per_run_time_limit=30,
         tmp_folder='/tmp/autoslearn_regression_example_tmp',
         output_folder='/tmp/autosklearn_regression_example_out')
     automl.fit(X_train, y_train, dataset_name='boston')