WIP: set n_classes as ndarray