Nice work! Here is an article you may find helpful if you have not already come across it.[0]. You may also want to consider benchmarking against some non ML methods.[1]
0. https://pubmed.ncbi.nlm.nih.gov/35318324/
1. https://www.nature.com/articles/s41586-023-06127-z