Instead, if you want the algorithm to dynamically select the class to which the model assigns the highest probability, set max_output (maybe max classd) to ON