Skip to content

Commit

Permalink
Fix bug with predictions from Lrnr_grf
Browse files Browse the repository at this point in the history
The .predict method for Lrnr_grf had an incorrect argument name for grf::predict.quantile_forest: the argument for supplying new data should be called "newdata", not "new_data" (see https://grf-labs.github.io/grf/reference/predict.quantile_forest.html).
  • Loading branch information
herbps10 authored Feb 13, 2023
1 parent 62f6454 commit 74f7538
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion R/Lrnr_grf.R
Original file line number Diff line number Diff line change
Expand Up @@ -125,7 +125,7 @@ Lrnr_grf <- R6Class(
# generate predictions and output
predictions_list <- stats::predict(
private$.fit_object,
new_data = data.frame(task$X),
newdata = data.frame(task$X),
quantiles = quantiles_pred
)
predictions <- as.numeric(predictions_list$predictions)
Expand Down

0 comments on commit 74f7538

Please sign in to comment.