Reducing bias in ML model training