2_2最近邻算法kNN(k_nearest_neighbor)——file2matrix_2_2

声明:《机器学习实战》代码详细注释和重构,以及相关的函数、模块和算法的解释;本文为博主原创文章,未经博主允许不得转载。

1.k-近邻算法(程序清单2-2)

*#代码:Peter Gong_shuai
*#中文注释:Gong_shuai
*#代码重构:Gong_shuai
*#函数注解:Gong_shuai
*#相关的函数、模块和算法的解释:Gong_shuai

2. 代码

#coding=utf-8
#k-近邻算法(程序清单2-2)

#源代码:Peter  Gong_shuai
#中文注释:Gong_shuai
#代码重构:Gong_shuai
#函数注解:Gong_shuai
#环境:python2.7

from numpy import *
import operator
from os import listdir #函数listdir可以列出给定目录的文件名



#处理输入格式问题,输入为文件名字符串,输出为训练样本矩阵和类标签向量
def file2matrix(filename):
    fr = open(filename)
    numberOfLines = len(fr.readlines())         #得到文本行数
    returnMat = zeros((numberOfLines,3))        #创建以零填充的矩阵,为了简化,另外的一个维度设为3
    classLabelVector = []                       #返回标签     
    fr = open(filename)    #解析文件数据到列表,循环处理文件中的每一行的数据
    index = 0
    for line in fr.readlines():
        line = line.strip()    #截取掉所有的回车字符
        listFromLine = line.split('\t')    #将整行数据分割成一个元素列表
        returnMat[index,:] = listFromLine[0:3]     #选取前三个元素,存储到特征矩阵中
        classLabelVector.append(int(listFromLine[-1]))
        index += 1
    return returnMat,classLabelVector

# datingDataMat,datingLabels = file2matrix('datingTestSet.txt')
# print datingDataMat
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章