File tree Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Expand file tree Collapse file tree 1 file changed +4
-5
lines changed Original file line number Diff line number Diff line change 1
1
test_that(" single-step surv" , {
2
- task = mlr_tasks $ get(" unemployment " )
2
+ task = mlr_tasks $ get(" gbcs " )
3
3
learner = mlr_learners $ get(" surv.kaplan" )
4
4
5
- train_task = task $ clone()$ filter(1 : 3000 )
5
+ train_task = task $ clone()$ filter(1 : 600 )
6
6
learner $ train(train_task )
7
- newdata = remove_named(task $ clone()$ filter(3001 : 3343 )$ data(), task $ target_names )
7
+ newdata = remove_named(task $ clone()$ filter(601 : 686 )$ data(), task $ target_names )
8
8
p = learner $ predict_newdata(task = train_task , newdata = newdata )
9
9
10
10
p = as.data.table(p )
11
- expect_data_table(p , nrow = 343L )
11
+ expect_data_table(p , nrow = 86L )
12
12
expect_true(allMissing(p $ time ))
13
13
expect_true(allMissing(p $ status ))
14
14
})
15
15
16
-
17
16
test_that(" single-step dens" , {
18
17
task = tsk(" faithful" )
19
18
learner = lrn(" dens.kde" )
You can’t perform that action at this time.
0 commit comments