JakeColtman / bartpy

Bayesian Additive Regression Trees For Python
https://jakecoltman.github.io/bartpy/
MIT License
219 stars 44 forks source link

What should be the shape of inputs, X and Y? #46

Open samsenyang opened 4 years ago

samsenyang commented 4 years ago

What should be the shape of inputs, X and Y? Because when X is with shape (4802, 79), which means 4802 samples and 79 covariates per sample, and Y is with shape (4802), there will be an error:

IndexError: too many indices for array

when I use SklearnModel.

Looking for answers.