Submitted on 7 July 2022
Comment: 9 pages, 7 figures. accepted to the ICML 2022 Machine Learning for Astrophysics workshop
Subjects: Astrophysics - Instrumentation and Methods for Astrophysics; Astrophysics - Astrophysics of Galaxies; Computer Science - Machine Learning