remove_missing_levels | R Documentation |
Accounts for missing factor levels present only in test data but not in train data by setting values to NA
remove_missing_levels(fit, test_data)
fit |
fitted model on training data |
test_data |
data to make predictions for |
data.frame with matching factor levels to fitted model
foo <- data.frame(
response = rnorm(3),
predictor = as.factor(c("A", "B", "C"))
)
model <- lm(response ~ predictor, foo)
foo.new <- data.frame(predictor = as.factor(c("A", "B", "C", "D")))
predict(model, newdata = remove_missing_levels(
fit = model,
test_data = foo.new
))
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.