Geothermal Battery Optimization Using Stochastic Hydro-Thermal Simulations and Machine Learning Algorithms