Fig. 4.

Simulated Treatment Log-Risk Surface. Treatment Log-Risk Surfaces as a function of a patient’s relevant covariates x₀ and x₁. a The true log-risk h₁(x) if all patients in the test set were given treatment τ=1. We then manually set all treatment groups to either τ=0 or τ=1. b The predicted log-risk ${ĥ}_0\left(x\right)$ for patients with treatment group τ=0. c The network’s predicted log-risk ${ĥ}_1\left(x\right)$ for patients in treatment group τ=1