中文字幕av专区_日韩电影在线播放_精品国产精品久久一区免费式_av在线免费观看网站

溫馨提示×

knn算法python代碼怎么寫

小億
84
2024-01-04 10:43:35
欄目: 編程語言

下面是一個簡單的KNN算法的Python代碼示例:

import numpy as np
from collections import Counter

def euclidean_distance(x1, x2):
    return np.sqrt(np.sum((x1 - x2) ** 2))

class KNN:
    def __init__(self, k=3):
        self.k = k
    
    def fit(self, X, y):
        self.X_train = X
        self.y_train = y
    
    def predict(self, X):
        y_pred = [self._predict(x) for x in X]
        return np.array(y_pred)
    
    def _predict(self, x):
        # 計算所有訓練樣本與待預測樣本的距離
        distances = [euclidean_distance(x, x_train) for x_train in self.X_train]
        # 根據距離排序并獲取前k個樣本的索引
        k_indices = np.argsort(distances)[:self.k]
        # 獲取前k個樣本的標簽
        k_labels = [self.y_train[i] for i in k_indices]
        # 返回出現次數最多的標簽作為預測結果
        most_common = Counter(k_labels).most_common(1)
        return most_common[0][0]

使用示例:

X_train = np.array([[1, 2], [1.5, 1.8], [5, 8], [8, 8], [1, 0.6], [9, 11]])
y_train = np.array([0, 0, 1, 1, 0, 1])

knn = KNN(k=3)
knn.fit(X_train, y_train)

X_test = np.array([[2, 3], [6, 9], [1, 1]])
y_pred = knn.predict(X_test)

print(y_pred)  # 輸出:[0, 1, 0]

這個示例中使用的是歐氏距離作為距離度量方法,同時實現了一個簡單的KNN類,其中的fit()方法用于訓練模型,predict()方法用于預測新樣本的標簽。KNN類的_predict()方法用于計算單個樣本的預測結果。

0
略阳县| 陇川县| 巴青县| 济南市| 石嘴山市| 辽阳市| 钟山县| 西华县| 蕉岭县| 太湖县| 大厂| 普宁市| 鄂托克前旗| 泰宁县| 蓝田县| 萝北县| 武宣县| 临潭县| 利辛县| 富顺县| 元谋县| 泰州市| 明水县| 东至县| 青田县| 横峰县| 宁津县| 姜堰市| 乐亭县| 洛浦县| 东安县| 射阳县| 禄劝| 萨迦县| 佛冈县| 芦溪县| 东阿县| 崇礼县| 元氏县| 东平县| 新营市|