I don’t want to convert the mxnet dataset to an ndarray and then manually filter out the elements not of interest.
First I’m relabeling with a transform function and then providing a custom batchify function. But the design of mxnet is making this painful, as the zipper only allows access to the labels after it has stacked the predictor variables into a big ndarray.
How do I write a batchify function to remove the unwanted data points?
Edit: I’m aware that this won’t work because there batches would differ in size after removing unwanted elements, but can’t think of a reasonable solution, so am giving up.