I have a problem in fitting BART model for classification in python. I have tried to follow https://github.com/JakeColtman/bartpy unfoturnately I get an error for multi-class outcome. Does anyone know how to implement BART for multiclass problems in python?
For example I try to obtain accuracy scores using balanced_accuracy_score(y_test, bart_model_pred) but gets an error
ValueError: Classification metrics can't handle a mix of multiclass and continuous targets