Как использовать пакет рецептов для замены пропущенных значений константой

Я не могу понять, как использовать пакет recipes, чтобы заменить отсутствующие числовые переменные константой.

Я действительно думал об использовании step_lowerimpute, но не думаю, что смогу использовать его для своего случая. step_lowerimpute заменяет отсутствующие значения ниже заданного порога случайными числами между 0 и порогом. В моем случае это не сработает.

Например, у меня есть некоторая лабораторная переменная, например, молочная кислота, которая часто отсутствует. Я хочу заменить отсутствующие значения экстремальным значением, например -9999.

1 ответ

Это мой первый день, глядя на пакет рецептов (так что, возможно, не самый надежный ответ...). У меня был такой же вопрос, и я считаю, что по мере необходимости работает следующее:

rec <-
    recipe( ~ ., data = airquality) %>%
    step_mutate(
        Ozone = tidyr::replace_na(Ozone, -9999)
    ) %>%
    prep(training = airquality, retain = TRUE)

juice(rec)

Прежде чем натолкнуться на этот метод, я также попытался создать свой собственный шаг, который также, кажется, работает, но выше намного проще...

step_nareplace <- 
    function(recipe, 
            ..., 
            role = NA, 
            trained = FALSE,  
            skip = FALSE,
            columns = NULL,
            replace = -9,
            id = rand_id("nareplace")) {
  add_step(
    recipe,
    step_nareplace_new(
      terms = ellipse_check(...),
      role = role,
      trained = trained,
      skip = skip,
      id = id,
      replace = replace,
      columns = columns
    )
  )
}

step_nareplace_new <- 
    function(terms, role, trained, skip, id, columns, replace) {
  step(
    subclass = "nareplace",
    terms = terms,
    role = role,
    trained = trained,
    skip = skip,
    id = id,
    columns = columns,
    replace = replace
  )
}

prep.step_nareplace <- function(x, training, info = NULL, ...) {

    col_names <- terms_select(x$terms, info = info)

    step_nareplace_new(
        terms = x$terms,
        role = x$role,
        trained = TRUE,
        skip = x$skip,
        id = x$id,
        columns = col_names,
        replace = x$replace
      )
}

bake.step_nareplace <- function(object, new_data, ...) {
  for (i in  object$columns) {
    if (any(is.na(new_data[, i])))
      new_data[is.na(new_data[, i]), i] <- object$replace
  }
  as_tibble(new_data)
}

print.step_nareplace <-
  function(x, width = max(20, options()$width - 30), ...) {
    cat("Replacing NA values in ", sep = "")
    cat(format_selectors(x$terms, wdth = width))
    cat("\n")
    invisible(x)
  }

tidy.step_nareplace <- function(x, ...) {
  res <- simple_terms(x, ...)
  res$id <- x$id
  res
}


recipe(Ozone ~ ., data = airquality) %>%
   step_nareplace(Ozone, replace = -9999) %>%
   prep(airquality, verbose = FALSE, retain = TRUE) %>%
   juice()

Вы можете попробовать использовать step_unknown() функция, заменяющая пропущенные значения NA значения с new_level что может предоставить пользователь.

Зачем вам конкретно нужен пакет рецептов для этого? Просто заменить все NA постоянным значением можно довольно просто.

library(imputeTS)
na.replace(yourDataframe, fill = -9999)

Другое решение (без дополнительной упаковки):

yourDataframe[is.na(yourDataframe)] <- -9999
Другие вопросы по тегам