test_that("Tests that random forest is working correctly", {
x <- iris[, -1]
y <- iris[, 1]
context('Forestry base function')
# Set seed for reproductivity
set.seed(24750371)
# Test forestry (mimic RF)
forest <- forestry(
x,
y,
ntree = 500,
replace = TRUE,
sample.fraction = .8,
mtry = 3,
nodesizeStrictSpl = 5,
nthread = 2,
splitrule = "variance",
splitratio = 1,
nodesizeStrictAvg = 5
)
# Test predict
y_pred <- predict(forest, x)
# Mean Square Error
sum((y_pred - y) ^ 2)
expect_equal(sum((y_pred - y) ^ 2), 9.84, tolerance = 0.5)
})
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.