Fix visualizer for classification
, mode=simple
#442
Merged
classification
, mode=simple
#442