Learn R Programming

grf (version 2.3.2)

get_scores.causal_survival_forest: Compute doubly robust scores for a causal survival forest.

Description

For details see section 3.2 in the causal survival forest paper.

Usage

# S3 method for causal_survival_forest
get_scores(forest, subset = NULL, num.trees.for.weights = 500, ...)

Value

A vector of scores.

Arguments

forest

A trained causal survival forest.

subset

Specifies subset of the training examples over which we estimate the ATE. WARNING: For valid statistical performance, the subset should be defined only using features Xi, not using the treatment Wi or the outcome Yi.

num.trees.for.weights

Number of trees used to estimate Var[W | X = x]. Note: this argument is only used in the case of a continuous treatment (see get_scores.causal_forest for details).

...

Additional arguments (currently ignored).