前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >kNN分类算法实例1:用kNN改进约会网

kNN分类算法实例1:用kNN改进约会网

作者头像
py3study
发布2020-01-16 17:39:44
1.8K0
发布2020-01-16 17:39:44
举报
文章被收录于专栏:python3python3

实战内容

海伦女士一直使用在线约会网站寻找适合自己的约会对象。尽管约会网站会推荐不同的人选,但她并不是喜欢每一个人。经过一番总结,她发现自己交往过的人可以进行如下分类:

代码语言:javascript
复制
不喜欢的人
魅力一般的人
极具魅力的人

海伦收集约会数据已经有了一段时间,她把这些数据存放在文本文件datingTestSet.txt中,每个样本数据占据一行,总共有1000行。

海伦收集的样本数据主要包含以下3种特征:

代码语言:javascript
复制
每年获得的飞行常客里程数
玩视频游戏所消耗时间百分比
每周消费的冰淇淋公升数

任务:试建立一个分类器,使得在下次输入数据后,程序可以帮助海伦预测海伦对此人的印象。

原著中,所有归一化、kNN算法,分类器都是作者自己写的。代码可以用于理解算法原理,用于使用就没有必要,而且代码基于的版本是2.7,难以直接使用。 源代码及其详解可以参考以下链接: 机器学习实战—k近邻算法(kNN)02-改进约会网站的配对效果

既然有了优秀的sklearn库可以为我们提供现成的kNN函数,为什么不直接调用它呢?这正是python较其他语言强大的所在呀!

用sklearn自带库实现kNN算法分类

大致流程:

  1. 导入数据,打印数据的相关信息,初步了解数据
  2. 绘制图像更直观的分析数据
  3. 切分数据成测试集和训练集,可以用sklearn自带库随机切割,也可以将数据前半部分和后半部分切割,后者更有利于代入测试集人工检验
  4. 数据预处理,之后的代码仅有归一化
  5. 用sklearn自带库训练算法,然后打分正确率
  6. 完善分类器功能,允许后期输入参数真正实现分类

可以参考以下链接,更详细的了解sklearn自带的kNN算法做分类的流程: 用sklearn实现knn算法的实现流程

以下是代码(更多细节请参考附在最后的参考资料):

代码语言:javascript
复制
#!/usr/bin/env python 
# -*- coding:utf-8 -*-
import numpy as np
import pandas as pd
import seaborn as sns
import matplotlib.pyplot as plt
from sklearn import preprocessing
from sklearn.model_selection import train_test_split
from sklearn.neighbors import KNeighborsClassifier

#转化文件格式 第一次执行后,在文件夹下会生成.csv文件,之后就不需要重复执行这段代码了
'''
txt = np.loadtxt('datingTestSet2.txt')
txtDf = pd.DataFrame(txt)
txtDf.to_csv('datingTestSet2.csv', index=False) #no index
'''

#load csv, learn more about it.
dataset = pd.read_csv('datingTestSet2.csv')
dataset.columns = ['miles', 'galons', 'percentage', 'label']
print(dataset.head())
print(dataset.dtypes)
print(np.unique(dataset['label']))
print(len(dataset))


#analyze our set through seaborn
# 绘制散点图 第一次执行后,三个特征对结果的影响就会有个印象,后面也可以不再执行
'''
sns.lmplot(x='galons', y='percentage', data=dataset, hue='label',fit_reg=False)
sns.lmplot(x='miles', y='percentage', data=dataset, hue='label',fit_reg=False)
sns.lmplot(x='miles', y='galons', data=dataset, hue='label',fit_reg=False)
plt.show()
'''


#cut dataset randomly
'''
dataset_data = dataset[['miles', 'galons', 'percentage']]
dataset_label = dataset['label']
print(dataset_data.head())
data_train, data_test, label_train, label_test = train_test_split(dataset_data, dataset_label, test_size=0.2, random_state=0)
'''
#cut dataset
dataset_data = dataset[['miles', 'galons', 'percentage']]
dataset_label = dataset['label']

data_train = dataset.loc[:800,['miles', 'galons', 'percentage']]  #我让训练集取前800个
print(data_train.head())
label_train = np.ravel(dataset.loc[:800,['label']])
data_test = dataset.loc[800:,['miles', 'galons', 'percentage']]
label_test = np.ravel(dataset.loc[800:,['label']])

#preprocessing, minmaxscaler
min_max_scaler = preprocessing.MinMaxScaler()
data_train_minmax = min_max_scaler.fit_transform(data_train)
data_test_minmax = min_max_scaler.fit_transform(data_test)
print(data_train_minmax)

#training and scoring
knn = KNeighborsClassifier(n_neighbors=15)
knn.fit(data_train_minmax,label_train)
score = knn.score(X=data_test_minmax,y=label_test,sample_weight=None)
print(score)

#completion
def classifyperson(): #此为手动输入参数预测结果需要的函数
    percentage = float(input('percentage of time spent playing video games?'))
    ffMiles = float(input('frequent flier miles earned per year?'))
    iceCream = float(input('liters of ice-cream consumed per year?'))
    inArr = np.array([[percentage, ffMiles, iceCream]])
    inArr_minmax = min_max_scaler.fit_transform(inArr)
    return inArr_minmax

#inArr_minmax = classifyperson() 

label_predict = knn.predict(data_test_minmax) #此代码与之前人工切分数据集结合,用于人工校对正确率
print(label_predict)

当k取15的时候,正确率试过来是最高的,能达到0.935 以下是测试集代入分类器后得到的结果,可以将其与文本文件里最后200个标签一一对照一下,可以发现正确率确实还是蛮高的。

KNN分类算法
KNN分类算法

将内含非数值型的txt文件转化为csv文件

原作中,作者已经将obj型标签帮我们转化成数值型了,因此在上面的代码中,我们可以直接将转化好的文件拿来用。但是如果要我们自己转化数据类型,该怎么转化? 其实只需要将原作中的第一个函数略加改造即可。代码如下:

代码语言:javascript
复制
# 将文本记录转换为NumPy的解析程序
def file2matrix(filename):
    fr = open(filename)
    #得到文件行数
    arrayOfLines = fr.readlines()
    numberOfLines = len(arrayOfLines)
    #创建返回的Numpy矩阵
    returnMat = np.zeros((numberOfLines,3))
    classLabelVector = []
    #解析文件数据到列表
    index = 0
    for line in arrayOfLines:
        line = line.strip() #注释1
        listFromLine = line.split('\t') #注释2
        returnMat[index,:] = listFromLine[0:3]
        classLabelVector.append(listFromLine[-1])
        index += 1
    return returnMat,classLabelVector

#调用函数,读取数据
datingDataMat,datingLabels = file2matrix('datingTestSet.txt')
#拼接标签和特征
datingDataMat_df = pd.DataFrame(datingDataMat)
datingLabels_df = pd.DataFrame(datingLabels)
txtDf = pd.concat([datingDataMat_df,datingLabels_df],axis=1) #横向拼接
txtDf.to_csv('datingTestSet.csv', index=False) #这里,datingTestSet.csv中的标签是[largedoses, smalldoses, didn't like]

#读取datingTestSet.csv
dataset = pd.read_csv('datingTestSet.csv')
dataset.columns = ['miles', 'galons', 'percentage', 'label']
print(dataset.head())

得到的新DataFrame如下:

kNN算法
kNN算法

用sns.lmplot绘图反映几个特征之间的关系

以下列出了三个特征两两之间的关系(没有列全),通过它们大致能感觉出三个特征值对结果的影响。

kNN分类算法
kNN分类算法
kNN算法
kNN算法
kNN算法
kNN算法

参考资料

如何把txt文件转化为csv格式? (此办法只适用于只有数值型的文件,或者说标签已经被转化为数值型了,如何将含object型的txt文件导入见后) 如何对DataFrame的列名重新命名? pycharm如何用run执行不用console执行? 如何绘制散点图? 如何改变DataFrame某一列的数据类型? 如何使用seaborn中的jointplot? 查看某一列有那些值? jointplot没有hue参数,有什么其他函数可以代替吗? 如何绘制子图? 如何获取Dataframe的行数和列数? 如何选取DataFrame列?官网 如何切分数据集?官网 如何用sklearn的train_test_split随机切分数据集? 如何用sklearn自带库归一化?官网 归一化、标准化、正则化介绍及实例 如何使用sklearn中的knn算法? 用sklearn实现knn算法的实现流程 洗牌函数shuffle()和permutation()的区别是什么? 如何使用with open()as filename? 如何用Python提取TXT数据转化为DataFrame? pandas dataframe的合并(append, merge, concat)

本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2019-05-29 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 实战内容
  • 用sklearn自带库实现kNN算法分类
  • 将内含非数值型的txt文件转化为csv文件
  • 用sns.lmplot绘图反映几个特征之间的关系
  • 参考资料
相关产品与服务
文件存储
文件存储(Cloud File Storage,CFS)为您提供安全可靠、可扩展的共享文件存储服务。文件存储可与腾讯云服务器、容器服务、批量计算等服务搭配使用,为多个计算节点提供容量和性能可弹性扩展的高性能共享存储。腾讯云文件存储的管理界面简单、易使用,可实现对现有应用的无缝集成;按实际用量付费,为您节约成本,简化 IT 运维工作。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档