targeted (version 0.9.0)

predict_filter_bound: Prediction filter bounding predictions to fixed range

Description

Generates a prediction filter for the predict.filter argument of learner that bounds predictions to a fixed range (lower, upper).

Usage

predict_filter_bound(lower = NULL, upper = NULL)

Value

A filter generator function (see learner).

Arguments

lower

(numeric) Lower bound, or NULL for no lower bound.

upper

(numeric) Upper bound, or NULL for no upper bound.

Author

Benedikt Sommer

Examples

data(cars)
lr <- learner_glm(
  speed ~ dist,
  learner.args = list(predict.filter = predict_filter_bound(upper = 10))
)
lr$estimate(cars)
lr$predict(data.frame(dist = c(10, 50)))