29 lines
867 B
R
29 lines
867 B
R
|
context("Make sure we can save forests while training")
|
||
|
|
|
||
|
|
test_that("Can save a random forest while training, and use it afterward", {
|
||
|
|
|
||
|
|
expect_false(file.exists("trees")) # Folder shouldn't exist yet
|
||
|
|
|
||
|
|
x1 <- rnorm(1000)
|
||
|
|
x2 <- rnorm(1000)
|
||
|
|
y <- 1 + x1 + x2 + rnorm(1000)
|
||
|
|
|
||
|
|
data <- data.frame(x1, x2, y)
|
||
|
|
forest <- train(y ~ x1 + x2, data,
|
||
|
|
ntree=100, numberOfSplits = 5, mtry = 1, nodeSize = 5,
|
||
|
|
savePath="trees")
|
||
|
|
|
||
|
|
expect_true(file.exists("trees")) # Something should have been saved
|
||
|
|
|
||
|
|
# try making a little prediction to verify it works
|
||
|
|
newData <- data.frame(x1=seq(from=-3, to=3, by=0.5), x2=0)
|
||
|
|
predictions <- predict(forest, newData)
|
||
|
|
|
||
|
|
# Also make sure we can load the forest too
|
||
|
|
newforest <- load_forest("trees")
|
||
|
|
predictions <- predict(newforest, newData)
|
||
|
|
|
||
|
|
|
||
|
|
unlink("trees", recursive=TRUE)
|
||
|
|
|
||
|
|
})
|