I’m not sure you can increase the probability of certain classes without training the model.
You could index these four classes and just use the argmax on these logits to avoid getting invalid classes.