IndexError: too many indices for array for scikit-learn

Question

When I run

import numpy as np
from sklearn.cluster import MeanShift, estimate_bandwidth
estimate_bandwidth(np.array([1,2,3,4,5,6]))

I get the error

Traceback (most recent call last):
  File "<input>", line 1, in <module>
  File "[..]/anaconda2/envs/reifen/lib/python2.7/site-packages/sklearn/cluster/mean_shift_.py", line 72, in estimate_bandwidth
    d, _ = nbrs.kneighbors(X[batch, :], return_distance=True)
IndexError: too many indices for array

I understand that the dimensionality of the array is not correct - but I am not sure what I did wrong here - the code isn't that long. Can estimate_bandwidth only be used with data that is multivariate?

You should read the documentation - scikit-learn.org/stable/modules/generated/… — Craicerjack
– Craicerjack, Commented May 12, 2017 at 10:03
@Craicerjack: I did, but didn't read anything about this. However reading again, I gathered that for the univariate case I need to use .reshape((6, 1)) after the array. Thanks. — Make42
– Make42, Commented May 12, 2017 at 10:08

Stavros Avramidis · Accepted Answer · 2017-05-12 10:11:03Z

1

Ypu should do something like this:

import numpy as np
from sklearn.cluster import MeanShift, estimate_bandwidth

array = np.array([1,2,3,4,5,6])

d = estimate_bandwidth(array.reshape(len(array),1))

answered May 12, 2017 at 10:11

Stavros Avramidis

8659 silver badges21 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Make42 Over a year ago

Although this is exactly what I just commented ;-) - it is the correct solution.

Make42 Over a year ago

Yeah. probably :-). Happens.

Collectives™ on Stack Overflow

IndexError: too many indices for array for scikit-learn

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related