diff --git a/dummy-vars-playground.R b/dummy-vars-playground.R index 715083b..f51a52f 100644 --- a/dummy-vars-playground.R +++ b/dummy-vars-playground.R @@ -8,10 +8,7 @@ rm(seqs) load(here("notebooks/data/nback_seqs.Rd")) -seqs <- seqs %>% - mutate(stimulus = lag(stimulus)) - -f <- as.formula("correct ~ stimulus + n") +f <- as.formula("correct ~ n + stimulus + sl") set.seed(42) @@ -61,7 +58,7 @@ tune <- expand.grid(alpha = 0:1, lambda = seq(0, 0.01, length = 100)) # pls tune -tune <- expand.grid(ncomp=1:5) +tune <- expand.grid(ncomp=1:20) model <- train(seqs.train.x, seqs.train.y,