Predicting near-field specific impulse distributions using machine learning