Skip to content

Commit

Permalink
less trees for efficiency
Browse files Browse the repository at this point in the history
  • Loading branch information
bcjaeger committed Oct 23, 2023
1 parent 754b431 commit 56944e0
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions vignettes/pd.Rmd
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ pbc_orsf_test <- pbc_orsf[-index_train, ]
fit <- orsf(data = pbc_orsf_train,
formula = Surv(time, status) ~ . - id,
n_tree = 50,
oobag_pred_horizon = pred_horizon)
fit
Expand Down Expand Up @@ -92,14 +93,15 @@ You can compute PD three ways with `aorsf`:

- new data PD shows how the model predicts outcomes for observations it has not seen. This is helpful if you want to test your model's reliability or fairness.

Let's re-fit our ORSF to all available data before proceeding to the next sections.
Let's re-fit our ORSF model to all available data before proceeding to the next sections.

```{r}
set.seed(329730)
fit <- orsf(pbc_orsf,
Surv(time, status) ~ . -id,
n_tree = 50,
oobag_pred_horizon = pred_horizon)
```
Expand Down Expand Up @@ -171,7 +173,7 @@ Now would it be tedious if you wanted to do this for all the variables? You bet.

```{r}
pd_smry <- orsf_summarize_uni(fit)
pd_smry <- orsf_summarize_uni(fit, n_variables = 4)
pd_smry
Expand Down

0 comments on commit 56944e0

Please sign in to comment.