|
@@ -15,7 +15,7 @@ Created on Wed Sep 30 14:15:17 2020
|
|
|
|
|
|
import os
|
|
|
import sys
|
|
|
-import time
|
|
|
+import datetime
|
|
|
from itertools import product
|
|
|
from collections import ChainMap
|
|
|
from sklearn.pipeline import Pipeline
|
|
@@ -141,7 +141,8 @@ class GridSearchPipelineSelector(PipelineSelector):
|
|
|
|
|
|
self.finished_tuning = True
|
|
|
|
|
|
- self.total_tuning_time = time.time() - self.start_tuning_time
|
|
|
+ self.total_tuning_time = datetime.datetime.today()\
|
|
|
+ - self.start_tuning_time
|
|
|
|
|
|
self._backup_trials()
|
|
|
|
|
@@ -344,6 +345,7 @@ if __name__ == "__main__":
|
|
|
gs.run_trials()
|
|
|
|
|
|
logger.info("Best trial: {}".format(gs.best_trial))
|
|
|
+ logger.info("Total tuning time: {}".format(gs.total_tuning_time))
|
|
|
|
|
|
for file in [trials_path, data_path, cv_path]:
|
|
|
os.remove(file)
|