lee-group-cmu / RFCDE

Random Forests for Conditional Density Estimation

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

oob_weights fails

colorlace opened this issue · comments

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")

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")