新人求助

jzandy629

使用pip现在的sklearn模块使用的时候显示里面的kmeans程序出错了，怎么办？



错误显示如下：

Traceback (most recent call last):
  File "D:\python\pythonc\Untitled.py", line 23, in <module>
    y_pred = KMeans(n_clusters=list1, random_state=9).fit_predict(X)
  File "D:\python\lib\site-packages\sklearn\cluster\_kmeans.py", line 1008, in fit_predict
    return self.fit(X, sample_weight=sample_weight).labels_
  File "D:\python\lib\site-packages\sklearn\cluster\_kmeans.py", line 861, in fit
    if _num_samples(X) < self.n_clusters:
TypeError: '<' not supported between instances of 'int' and 'range'

jzandy629

求助
啊

BngThea

KMeans的n_clusters参数要传入的是一个整数，而不是列表

Hoiste

TypeError的话看看是不是拿不同类型的数据进行比较了。