support for multiclass output prob