Skip to content

Commit 1bdebe9

Browse files
committed
MAINT add helper function for NIPS 2015 experiments
1 parent 055f5c0 commit 1bdebe9

File tree

1 file changed

+14
-2
lines changed

1 file changed

+14
-2
lines changed

autosklearn/automl.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -192,6 +192,18 @@ def fit_automl_dataset(self, dataset):
192192

193193
return self._fit(loaded_data_manager)
194194

195+
def fit_on_datamanager(self, datamanager):
196+
self._stopwatch = StopWatch()
197+
self._backend.save_start_time(self._seed)
198+
199+
name = os.path.basename(datamanager.name)
200+
self._stopwatch.start_task(name)
201+
self._start_task(self._stopwatch, name)
202+
self._dataset_name = name
203+
204+
self._logger = self._get_logger(name)
205+
self._fit(datamanager)
206+
195207
def _get_logger(self, name):
196208
logger_name = 'AutoML(%d):%s' % (self._seed, name)
197209
setup_logger(os.path.join(self._backend.temporary_directory, '%s.log' % str(logger_name)))
@@ -278,8 +290,8 @@ def _fit(self, datamanager):
278290
self._backend._make_internals_directory()
279291
if self._keep_models:
280292
try:
281-
os.mkdir(self._backend.get_model_dir())
282-
except OSError:
293+
os.makedirs(self._backend.get_model_dir())
294+
except (OSError, FileExistsError) as e:
283295
if not self._shared_mode:
284296
raise
285297

0 commit comments

Comments
 (0)