For choosing between 2 classes this is OK:
Don’t forget to connect the return value in your pic! And if you need more than 2:
But you do not get the weights. Admittedly there are numerous way to generate random results.