Using ML for feature normalization