Efficient stratification based on non-parametric regression methods