Skip to content

Commit cab211b

Browse files
committed
fix #208: mappend user time to milli seconds
1 parent 49b2d38 commit cab211b

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

openml/runs/functions.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ def _run_task_get_arffcontent(model, task, class_labels):
164164
model_fold.fit(trainX, trainY)
165165

166166
if version_complies(3, 3):
167-
modelfit_duration = time.process_time() - modelfit_starttime
167+
modelfit_duration = (time.process_time() - modelfit_starttime) * 1000
168168
user_defined_measures['usercpu_time_millis_training'][rep_no][fold_no] = modelfit_duration
169169

170170
if isinstance(model_fold, BaseSearchCV):
@@ -181,7 +181,7 @@ def _run_task_get_arffcontent(model, task, class_labels):
181181
ProbaY = model_fold.predict_proba(testX)
182182
PredY = model_fold.predict(testX)
183183
if version_complies(3, 3):
184-
modelpredict_duration = time.process_time() - modelpredict_starttime
184+
modelpredict_duration = (time.process_time() - modelpredict_starttime) * 1000
185185
user_defined_measures['usercpu_time_millis_testing'][rep_no][fold_no] = modelpredict_duration
186186
user_defined_measures['usercpu_time_millis'][rep_no][fold_no] = modelfit_duration + modelpredict_duration
187187

0 commit comments

Comments
 (0)