Fig. 2: UMAP visualization of the patients.

a UMAP visualization on the original features. b UMAP visualization on the representations obtained by the GNN encoder. Individual data points represent patients and color scale represent OS times (log days). UMAP uniform manifold approximation and projection, GNN graph neural network. Source data are provided with this paper.