Skip to content
This repository has been archived by the owner on Jul 27, 2018. It is now read-only.

fixed IndexError that arises by using c.classes_ as np index in base.py #28

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

whitmera
Copy link

@whitmera whitmera commented Jun 2, 2017

Brew threw an Index Error at line 189 in base.py.
187 if mode == 'probs':
188 probas = np.zeros((X.shape[0], n_classes))
--> 189 probas[:, list(c.classes_)] = c.predict_proba(X)
190 out[:, :, i] = probas
191

IndexError: only integers, slices (:), ellipsis (...), numpy.newaxis (None) and integer or boolean arrays are valid indices

It looks like it i trying to use floats as column indices for probas. this could be resolved by making change list(c.classes_) to list(c.classes.astype(int)) but I don't see the need to pre-create probas.

…s not ints, as a column index for a numpy array
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant