Surrogate modelling of transient stratified flow in a u-shaped pipe