机器学习练习：约会网站数据使用KNN分类

最新推荐文章于 2020-10-17 00:23:25 发布

Mr. Donkey_K

最新推荐文章于 2020-10-17 00:23:25 发布

阅读量600

点赞数

分类专栏：机器学习练习文章标签： KNN 机器学习分类约会网站

本文链接：https://blog.csdn.net/Dreaming5498/article/details/98481017

版权

本文基于《机器学习实战》一书，介绍如何利用K近邻（KNN）算法对约会网站的数据进行分类分析。通过对原始数据的处理和测试，展示了KNN在实战中的应用。

摘要由CSDN通过智能技术生成

代码来源：《机器学习实战》

# encoding=utf-8
"""
Date:2019-07-29 10:54
User:LiYu
Email:liyu_5498@163.com

"""
from numpy import *


def file2matrix(filename):
    """读取并处理数据"""
    with open(filename, 'r') as f:
        linesList = f.readlines()

    numberOfLines = len(linesList)
    returnMat = zeros((numberOfLines, 3))  # 1000*3的全0矩阵
    classLabelVector = []
    index = 0
    for line in linesList:
        line = line.strip().split('\t')
        # print(line[0:3])  # ['35018', '9.959588', '0.060020']
        # print(line[-1])  # largeDoses
        returnMat[index, :] = line[0:3]
        classLabelVector.append(line[-1])
        index += 1
    # print(numberOfLines) # 1000