Skip to content

Commit 38ca4b3

Browse files
committed
Fix ML vignette: cf_auc cross_fit not yet supported
1 parent 3f7981d commit 38ca4b3

1 file changed

Lines changed: 9 additions & 7 deletions

File tree

vignettes/ml-integration.Rmd

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -260,14 +260,15 @@ cf_mse(
260260
)
261261
```
262262

263-
## Works with All Estimators
263+
## Current Support
264264

265-
ML learners work with `cf_mse()`, `cf_auc()`, `cf_calibration()`, and their
266-
transportability variants (`tr_mse()`, `tr_auc()`, `tr_calibration()`):
265+
ML learners with cross-fitting are currently fully supported in `cf_mse()`.
266+
Support for `cf_auc()`, `cf_calibration()`, and the transportability variants
267+
is planned for a future release.
267268

268-
```{r auc-example, eval=requireNamespace("ranger", quietly = TRUE)}
269-
# AUC with ML nuisance models
270-
result_auc <- cf_auc(
269+
```{r mse-final-example, eval=requireNamespace("ranger", quietly = TRUE)}
270+
# Full example with ML nuisance models
271+
result_ml <- cf_mse(
271272
predictions = pred,
272273
outcomes = y,
273274
treatment = a,
@@ -277,10 +278,11 @@ result_auc <- cf_auc(
277278
propensity_model = ml_learner("ranger", num.trees = 100),
278279
outcome_model = ml_learner("ranger", num.trees = 100),
279280
cross_fit = TRUE,
281+
n_folds = 5,
280282
se_method = "influence"
281283
)
282284
283-
print(result_auc)
285+
print(result_ml)
284286
```
285287

286288
## Best Practices

0 commit comments

Comments
 (0)