mlr-org / mlr3

mlr3: Machine Learning in R - next generation
https://mlr3.mlr-org.com
GNU Lesser General Public License v3.0
914 stars 84 forks source link

predict_time can be (kind of) wrong #992

Open sebffischer opened 6 months ago

sebffischer commented 6 months ago

When calling learner_predict(), the predict_time in the state is overwritten. When doing a resampling and making predictions for different sets (in my case the holdout set and the test set), the time_train is not (as I expected) the total prediction time, but only the prediction time for one of the predictions.

It would be great to have access to the different prediction times.