Fig. 2: Comparison of survival predictions by our deep learning model and a Cox proportional-hazard model.

The broken lines around the KM curves represents 95% confidence intervals. Each baseline (0, 24, 48, and 72 h) has its own curve, in a separate panel. The predictions for both models are close to the Kaplan-Meier estimate for the full hold-out test set.