add option to get the feature selection results without saving the whole model to disk