DBSCAN聚類算法 Python 代碼


一、前言

二、DBSCAN聚類算法

三、參數選擇

四、DBSCAN算法迭代可視化展示

五、常用的評估方法:輪廓系數

六、用Python實現DBSCAN聚類算法

一、前言

去年學聚類算法的R語言的時候,有層次聚類、系統聚類、K-means聚類、K中心聚類,最后呢,被DBSCAN聚類算法迷上了,為什么呢,首先它可以發現任何形狀的簇,其次我認為它的理論也是比較簡單易懂的。今年在python這門語言上我打算好好弄弄DBSCAN。下面貼上它的官方解釋:

DBSCAN(Density-Based Spatial Clustering of Applications with Noise,具有噪聲的基於密度的聚類方法)是一種基於密度的空間聚類算法。 該算法將具有足夠密度的區域划分為簇,並在具有噪聲的空間數據庫中發現任意形狀的簇,它將簇定義為密度相連的點的最大集合。

二、DBSCAN聚類算法

文字看不懂看下面這個圖。下面這些點是分布在樣本空間的眾多樣本,現在我們的目標是把這些在樣本空間中距離相近的聚成一類。我們發現A點附近的點密度較大,紅色的圓圈根據一定的規則在這里滾啊滾,最終收納了A附近的5個點,標記為紅色也就是定為同一個簇。其它沒有被收納的根據一樣的規則成簇。(形象來說,我們可以認為這是系統在眾多樣本點中隨機選中一個,圍繞這個被選中的樣本點畫一個圓,規定這個圓的半徑以及圓內最少包含的樣本點,如果在指定半徑內有足夠多的樣本點在內,那么這個圓圈的圓心就轉移到這個內部樣本點,繼續去圈附近其它的樣本點,類似傳銷一樣,繼續去發展下線。等到這個滾來滾去的圈發現所圈住的樣本點數量少於預先指定的值,就停止了。那么我們稱最開始那個點為核心點,如A,停下來的那個點為邊界點,如B、C,沒得滾的那個點為離群點,如N)。

基於密度這點有什么好處呢,我們知道kmeans聚類算法只能處理球形的簇,也就是一個聚成實心的團(這是因為算法本身計算平均距離的局限)。但往往現實中還會有各種形狀,比如下面兩張圖,環形和不規則形,這個時候,那些傳統的聚類算法顯然就悲劇了。於是就思考,樣本密度大的成一類唄。吶這就是DBSCAN聚類算法。

三、參數選擇

上面提到了紅色圓圈滾啊滾的過程,這個過程就包括了DBSCAN算法的兩個參數,這兩個參數比較難指定,公認的指定方法簡單說一下:

  • 半徑:半徑是最難指定的 ,大了,圈住的就多了,簇的個數就少了;反之,簇的個數就多了,這對我們最后的結果是有影響的。我們這個時候K距離可以幫助我們來設定半徑r,也就是要找到突變點,比如:

以上雖然是一個可取的方式,但是有時候比較麻煩 ,大部分還是都試一試進行觀察,用k距離需要做大量實驗來觀察,很難一次性把這些值都選准。 

  • MinPts:這個參數就是圈住的點的個數,也相當於是一個密度,一般這個值都是偏小一些,然后進行多次嘗試

 

四、DBSCAN算法迭代可視化展示

國外有一個特別有意思的網站:https://www.naftaliharris.com/blog/visualizing-dbscan-clustering/

它可以把我們DBSCAN的迭代過程動態圖畫出來

設置好參數,點擊GO! 就開始聚類了!

直接跳到最后看一下DBSCAN的聚類結果,如下:

如果minPoints參數設置再大一點,那么這個笑臉可能會更好看。沒有顏色標注的就是圈不到的樣本點,也就是離群點,DBSCAN聚類算法在檢測離群點的任務上也有較好的效果。如果是傳統的Kmeans聚類,我們也來看一下效果:

是不是好丑,這完美的體現出來DBSCAN算法基於密度聚類的優勢了啊.

 

五、常用的評估方法:輪廓系數

這里提一下聚類算法中最常用的評估方法——輪廓系數(Silhouette Coefficient):

ji

  1. 計算樣本i到同簇其它樣本到平均距離ai。ai越小,說明樣本i越應該被聚類到該簇(將ai稱為樣本i到簇內不相似度)。
  2. 計算樣本i到其它某簇Cj的所有樣本的平均距離bij,稱為樣本i與簇Cj的不相似度。定義為樣本i的簇間不相似度:bi=min(bi1,bi2,...,bik2)
  • si接近1,則說明樣本i聚類合理
  • si接近-1,則說明樣本i更應該分類到另外的簇
  • 若si近似為0,則說明樣本i在兩個簇的邊界上

 六、用Python實現DBSCAN聚類算法

import pandas as pd
# 導入數據
beer = pd.read_csv('data.txt', sep=' ')
print(beer)

輸出結果:

from sklearn.cluster import DBSCAN
 
X = beer[["calories","sodium","alcohol","cost"]]
# 設置半徑為10,最小樣本量為2,建模
db = DBSCAN(eps=10, min_samples=2).fit(X)
 
labels = db.labels_ 
beer['cluster_db'] = labels  # 在數據集最后一列加上經過DBSCAN聚類后的結果
beer.sort_values('cluster_db')
 
# 注:cluster列是kmeans聚成3類的結果;cluster2列是kmeans聚類成2類的結果;scaled_cluster列是kmeans聚類成3類的結果(經過了數據標准化)

輸出結果:

# 查看根據DBSCAN聚類后的分組統計結果(均值)
print(beer.groupby('cluster_db').mean())

# 畫出在不同兩個指標下樣本的分布情況
print(pd.scatter_matrix(X, c=colors[beer.cluster_db], figsize=(10,10), s=100))

# 我們可以從上面這個圖里觀察聚類效果的好壞,但是當數據量很大,或者指標很多的時候,觀察起來就會非常麻煩。
from sklearn import metrics  
# 就是下面這個函數可以計算輪廓系數(sklearn真是一個強大的包)
score = metrics.silhouette_score(X,beer.cluster_db) 
print(score)

 

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM