Add n-fold batch and add data_range support in the model features