A data-driven method for modelling dissipation rates in stratified turbulence