Estimate training time
Arguments
- object
an untrained
aorsf
object- n_tree_subset
(integer) how many trees should be fit in order to estimate the time needed to train
object
. The default value is 10% of the trees specified inobject
. I.e., ifobject
hasn_tree
of 500, then the default valuen_tree_subset
is 50.
Value
a difftime object.
Examples
# specify but do not train the model by setting no_fit = TRUE.
object <- orsf(pbc_orsf, Surv(time, status) ~ . - id,
n_tree = 10, no_fit = TRUE)
# approximate the time it will take to grow 10 trees
time_estimated <- orsf_time_to_train(object, n_tree_subset=1)
print(time_estimated)
#> Time difference of 0.04550219 secs
# let's see how close the approximation was
time_true_start <- Sys.time()
orsf_train(object)
time_true_stop <- Sys.time()
time_true <- time_true_stop - time_true_start
print(time_true)
#> Time difference of 0.04227424 secs
# error
abs(time_true - time_estimated)
#> Time difference of 0.003227949 secs