Skip to content

Commit

Permalink
fix could not find function data()
Browse files Browse the repository at this point in the history
  • Loading branch information
cregouby committed Jun 15, 2024
1 parent b5fdfd6 commit 8ee3a93
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/testthat/setup.R
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@ suppressPackageStartupMessages(library(data.tree))


# ames small data
data("ames", package = "modeldata")
utils::data("ames", package = "modeldata")
ids <- sample(nrow(ames), 256)
small_ames <- ames[ids,]
x <- ames[ids,-which(names(ames) == "Sale_Price")]
Expand All @@ -20,7 +20,7 @@ ames_fit_vsplit <- tabnet_fit(x, y, tabnet_model=ames_pretrain_vsplit, epochs =
num_steps = 1, attention_width = 1, num_shared = 1, num_independent = 1)

# attrition small data
data("attrition", package = "modeldata")
utils::data("attrition", package = "modeldata")
ids <- sample(nrow(attrition), 256)

# attrition common models
Expand All @@ -35,7 +35,7 @@ attr_fitted <- tabnet_fit(attrix, attriy, epochs = 12)
attr_fitted_vsplit <- tabnet_fit(attrix, attriy, epochs = 12, valid_split=0.3)

# data.tree Node dataset
data("acme", package = "data.tree")
utils::data("acme", package = "data.tree")
acme_df <- data.tree::ToDataFrameTypeCol(acme, acme$attributesAll) %>%
select(-starts_with("level_"))

Expand Down

0 comments on commit 8ee3a93

Please sign in to comment.