search

adebayoj / fairml

Other
359 stars 74 forks source link

ValueError: cannot reshape array of size 34712 into shape (17356,) #13

Open Dola47 opened 4 years ago

Dola47 commented 4 years ago

Hi @adebayoj

I just tried to use Fairml to explain my XGBoost Model, and I got the aforementioned error.

I think the error is logical, because in the mse function, I think you are doing the reshape as if it was only a one class situation. However, for my case the output_constant_col and the normal_black_box_output is of size (17356,2). So I am working in a 2 class classification scenario.

Any ideas how to overcome the error mentioned?

Capture (1)

Thanks.

adebayoj commented 4 years ago

Hi @Dola47, thanks for raising this issue. I'll double check your example. I am currently in the midst of a deadline right now, so I'll get to this in about a week.

Overall, this library is going to undergo an overhaul over the summer, so keep an eye out.

Thanks!