I have succeeded to train a simple neural net to predict k=3 labels (original dataset is imbalanced, I downsampled the dominant class to make it somewhat balanced, and use weights in the loss function during training).
I want to use this trained network for inference on a new dataset of which I know the real distribution of (imbalanced) labels.
How can I use this knowledge of the distribution of labels in the new dataset I want to infer on?