Sum taken over wrong axis

[bchugg]

Hi there,

I believe the centered predictions are being computed incorrectly. Line 278 in forestci.py takes the average over the predictions, as opposed to the trees. The resulting shape of pred_mean is (forest.n_estimators,) when it should be (X_test.shape[0],). See below:

forest-confidence-interval/forestci/forestci.py

Line 278 in 6d2a9c2

pred_centered = pred - pred_mean

Thanks for the great package otherwise! :)

[nikaashpuri]

are there any thoughts on this one?

[nikaashpuri]

please check #80

[danieleongari]

pred_mean is correct to compute the variance of pred. I guess you missed that the matrix pred is trasposed.