Hi there, I am working on the [Intel Image Classification] (https://www.kaggle.com/puneet6060/intel-image-classification/kernels 2) dataset. It has six different classes: glacier, forest, street, building, mountain, and sea in it. However, I am only interested in training in some of the class. I only want to build a model to classify if an image is a forest or not forest. I can successfully load the data with dataloader function. Is there a way to override other five classes label into unforest and the other one stays as forest?