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 \(\hat {h}_{0}(x)\) for patients with treatment group τ=0. c The network’s predicted log-risk \(\hat {h}_{1}(x)\) for patients in treatment group τ=1