Update: The problem I have now is more specifically with the the Subset class.
In my CSV Class, I can get the distribution of my target easily by:
    def __len_target__(self):
        return np.bincount(self.data_frame.target)

The problem is, this method is inoperable when I use Subclass and torch.utils.data.random_split to lazily load the data and generate training, validation, and test sets.