Generalized Non-Linear Eddy Viscosity Models for Data-Assisted Reynolds Stress Closure