1. 程式人生 > >KNN分類演算法優缺點

KNN分類演算法優缺點

KNN演算法的優點:

1、思想簡單,理論成熟,既可以用來做分類也可以用來做迴歸;
2、可用於非線性分類;
3、訓練時間複雜度為O(n);
4、準確度高,對資料沒有假設,對outlier不敏感;

缺點:
1、計算量大;
2、樣本不平衡問題(即有些類別的樣本數量很多,而其它樣本的數量很少);
3、需要大量的記憶體;

其虛擬碼如下:
1. 計算已知類別資料集中的點與當前點之間的距離;
2. 按照距離遞增次序排序;
3. 選擇與當前距離最小的k個點;
4. 確定前k個點所在類別的出現概率
5. 返回前k個點出現頻率最高的類別作為當前點的預測分類。