Machine learning-augmented turbulence models for the simulation of two-phase shear flows