Multi-class BART Model Assistance

We have recently added a shape/size argument to BART. You can do something like:

    μ = pmx.BART("μ", X= X_train, Y= y_train, m= 50, shape=(num_categories, num_observations))