use stratified sampling for pre-processing