oob_weights fails
colorlace opened this issue · comments
Noah Fatsi commented
Running oob_weights gives me an error:
TypeError: zeros() takes at most 3 arguments (4 given)
I think the issue is with line 116 in ForestWrapper.pyx
:
wt_mat = np.zeros(self.n_train, self.n_train, dtype=int, order="F")
Noah Fatsi commented
Without looking too deeply it looks like maybe the extra argument is merely a typo and the line should read:
wt_mat = np.zeros(self.n_train, dtype=int, order="F")