Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Next Next commit
[skip ci] addressing #248
  • Loading branch information
Neeratyoy committed Feb 23, 2021
commit c3bc8a5f8f9e46c4d72eba2d6f361aca145d39c4
2 changes: 2 additions & 0 deletions openml/extensions/sklearn/extension.py
Original file line number Diff line number Diff line change
Expand Up @@ -1744,6 +1744,8 @@ def _prediction_to_probabilities(
user_defined_measures["usercpu_time_millis_training"] = modelfit_dur_cputime

modelfit_dur_walltime = (time.time() - modelfit_start_walltime) * 1000
if hasattr(model_copy, "refit_time_"):
modelfit_dur_walltime += model_copy.refit_time_
if can_measure_wallclocktime:
user_defined_measures["wall_clock_time_millis_training"] = modelfit_dur_walltime

Expand Down