Test PCA instead UMAP

[keckelt]

to project the probabilitites, a linear approach like PCA might make more sense

[keckelt]

Mustatea, S., Aupetit, M., Peltonen, J., Lespinats, S., & Dutykh, D. Supervised dimensionality reduction technique accounting for soft classes.
https://www.esann.org/sites/default/files/proceedings/2022/ES2022-26.pdf