《机器学习实战》代码实现学习一 使用K-近邻算法改进约会网站的配对效果(数据准备)

Posted 2021-02-09 chongchong0207

1.数据准备：从文本文件中解析数据

文本文件datingTestSet2.txt网盘地址为：

https://pan.baidu.com/s/19HNwo1TSWjWhbRwsyL-itg

提取码为：mz11

约会数据由1000行，主要包含一下三种特征：

每年获得的飞行常客里程数

玩视频游戏所耗时间百分比

每周消费的冰淇淋公升数

但是在把这些特征输入到分类器之前，必须将待处理数据格式改变为分类器可以接受的格式，在kNN.py中创建名为file2matrix的函数，以此来处理输入格式问题，该函数的输入为文件名字符串，输出为训练样本矩阵和类标签向量。

将文本记录转换为NumPy的解析程序