Data-driven models are an attractive alternative to reservoir simulation in workflows where full field-scale simulations may be computationally prohibitive [3,4]. One example is the forecasting and schedule optimization of waterflooding scenarios, where numerous function evaluations that correspond...