i have gotten this far:
class mydataset(Dataset):
    def __init__(self, features, labels):
        self.features = features
        self.labels = labels
        
    def __len__(self):
        return len(self.features)
    
    def __getitem__(self, index):
        image = self.features[idx]
        image = image.numpy()
        label = self.labels[idx]
        if label == 1:
            return image, label
        elif label == 7:
            return image, label

i’m not sure the last bit is right (the if statement)
and it states that image does not have a numpy() attribute