机器学习系列算法1：KNN

思路：空间上距离相近的点具有相似的特征属性。

执行流程：

　　•1. 从训练集合中获取K个离待预测样本距离最近的样本数据;
　　•2. 根据获取得到的K个样本数据来预测当前待预测样本的目标属性值

三要素：K值选择/距离度量（欧式距离）/决策选择（平均值/加权平均）

Knn问题：数据量大，计算量较大；解决方案：kd-tree

kd-tree：计算方差，根据方差大的划分

伪代码实现：

import numpy as np
from collections import defaultdict


class myknn:

    def fit(self, X, Y, k):
        self.train_x = X
        self.train_y = Y
        self.k = k

    def predict(self, X):
        predict_labels = []
        for x in X:
            # 1. 从训练数据中获取K个和当前待预测样本x最相似的样本
            neighbors = self.fetch_k_neighbors(x)
            # 2. 将这K个最相似的样本中出现次数最多的类别作为预测值
            predict_label = self.calc_max_count_label(neighbors)
            # 3. 将当前样本的预测值添加到临时的列表中
            predict_labels.append(predict_label)

        return predict_labels

    def fetch_k_neighbors(self, x):
        distances = []
        for neighbor in self.train_x:
            dis = np.sqrt(np.sum((np.array(x) - neighbor) ** 2))
            distances.append(dis)
        neighbors_y_distances = [[neighbor, y, dis] for neighbor, y, dis in zip(self.train_x, self.train_y, distances)]

        k_neighbors_y = sorted(neighbors_y_distances, key=lambda x: x[2])[:self.k]
        return k_neighbors_y

    def calc_max_count_label(self, neighbors):
        y_count = defaultdict(int)
        for neighbor, y, _ in neighbors:
            y_count[y] += 1
        max_count_label = sorted(y_count.items(), key=lambda x: x[1], reverse=True)[0][0]
        return max_count_label


a = myknn()
X = [
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9],
    [10, 11, 12],
    [1, 2, 3],
    [4, 5, 6],
    [7, 8, 9],
]
Y = [1, 2, 3, 2, 1, 2, 3]
k = 3
a.fit(X, Y, k)
print(a.predict([[7, 8, 9], ]))

sklearn 实现：

import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import matplotlib as mpl
import os

if not os.path.exists('models'): 
    os.mkdir('models')
base_path = './models/'  #模型保存路径

from sklearn.datasets import load_iris    #load 鸢尾花数据
from sklearn.model_selection import train_test_split  #数据分割
from sklearn.neighbors import KNeighborsClassifier  #knn分类器
from sklearn.externals import joblib   #持久化

data = pd.read_csv('iris.data', header=None)

print(data.head())
X = data.loc[:, :3] #前4列为特征数据
Y = data[4]  #最后一列为目标数据  
print(X.head())

X_train, X_test, Y_train, Y_test = train_test_split(X, Y, test_size=0.2, random_state=22)

'''
    def __init__(self,
            n_neighbors=5,   #邻居数目
            weights='uniform',  #uniform 等权重/distance
            algorithm='auto', #暴力计算和kd-tree
            leaf_size=30,  #构建kd-tree 允许的最大叶子节点
            p=2,    #距离公式
            metric='minkowski', #默认欧式距离
            metric_params=None,
            n_jobs=None,  #开启多少个线程计算
            **kwargs):'''

algo = KNeighborsClassifier(n_neighbors=3)
algo.fit(X_train, Y_train)
print('准确率：{}'.format(algo.score(X_train, Y_train)))

# 持久化
joblib.dump(value=algo, filename=base_path + 'knn.pkl')

机器学习开发流程：

# 1. 数据加载

# 2. 数据的清洗、处理

# 3. 训练数据和测试数据的划分

# 4. 特征工程

# 5. 模型对象构建

# 6. 模型训练

# 7. 模型效果评估

# 8. 模型持久化

持久化的方式主要三种：
-1. 将模型持久化为二进制的磁盘文件。
-2. 将模型参数持久化到数据库中。
-3. 使用模型对所有数据进行预测，并将预测结果保存到数据库中。

机器学习系列算法1：KNN

猜你喜欢