I just did a promote to variable from the GetSenseClassForStimulus node, made this an array and filled the desired values.
You could also just use the GetDisplayName from this as in my image and do a Switch On String. But I believe, that the other approach is better than dealing with strings.