Skip to content

Commit d1b3ebb

Browse files
committed
#103 - fix step_log_interval offset issue
1 parent d35c9f9 commit d1b3ebb

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

R/recipes-step_log_interval.R

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,8 @@ prep.step_log_interval <- function(x, training, info = NULL, ...) {
152152
limit_lower_trained <- training[, col_names] %>%
153153
purrr::map(.f = function(vals) {
154154

155+
vals <- vals + x$offset
156+
155157
max_x <- max(vals)
156158
min_x <- min(vals)
157159
range_x <- abs(max_x - min_x)
@@ -162,6 +164,8 @@ prep.step_log_interval <- function(x, training, info = NULL, ...) {
162164
limit_upper_trained <- training[, col_names] %>%
163165
purrr::map(.f = function(vals) {
164166

167+
vals <- vals + x$offset
168+
165169
max_x <- max(vals)
166170
min_x <- min(vals)
167171
range_x <- abs(max_x - min_x)
@@ -194,6 +198,9 @@ bake.step_log_interval <- function(object, new_data, ...) {
194198

195199
for (i in seq_along(object$limit_lower_trained)) {
196200

201+
print(object$limit_lower_trained[i])
202+
print(object$limit_upper_trained[i])
203+
197204
new_data[, param[i]] <- log_interval_vec(
198205
x = new_data %>% purrr::pluck(param[i]),
199206
limit_lower = as.numeric(object$limit_lower_trained[i]),

0 commit comments

Comments
 (0)