無法在 python function 內使用 Kmeans 集群

Question

如上所述 - 我正在嘗試創建一個 function 用於聚類特定數據類型並顯示它們。

function目前看起來有點像這樣，

def cluster(inputData):
    variable_s= inputData.groupby(['x','z', 'c'])['w'].sum().unstack()
    ## 4 Clusters
    model = cluster.MiniBatchKMeans(n_clusters=5)
    model.fit(variable_s.fillna(0))
    variable_s['kmeans_4'] = model.predict(variable_s.fillna(0))

    ## 8 Clusters
    model = cluster.KMeans(n_clusters=8)
    model.fit(variable_s.fillna(0))
    variable_s['kmeans_8'] = model.predict(variable_s.fillna(0))

    ## Looking at hourly distribution.
    variable_s_Hourly = variable_s.reset_index(1, inplace=True)
    variable_s_Hourly['hour'] = variable_s_Hourly.index.hour
    return variable_s, variable_s_Hourly

它用

from sklearn import cluster
    

進行聚類，它給了我這樣的錯誤，

AttributeError: 'function' object has no attribute 'MiniBatchKMeans'

關於解決這個問題的任何線索？ 我原以為 function 只要將庫導入文件本身就可以了-這是在jupyter筆記本中:)

干杯!

Answer 1

function 名稱（“集群”）會影響導入。 更改function名稱即可解決。

或者，您可以為導入指定別名：

from sklearn import cluster as clstr