File tree Expand file tree Collapse file tree 1 file changed +14
-2
lines changed Expand file tree Collapse file tree 1 file changed +14
-2
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments