knn算法实现电影分类

KNN英文全称K-nearst neighbor，中文名称为K近邻算法，它是由Cover和Hart在1968年提出来的
KNN算法原理：

计算已知类别数据集中的点与当前点之间的距离；
按照距离递增次序排序；
选择与当前距离最小的k个点；
确定前k个点所在类别的出现概率
返回前k个点出现频率最高的类别作为当前点的预测分

#-*-coding:utf-8-*-
import numpy  as  np
import pandas  as pd
def distance(v1,v2):dist=np.sqrt(np.sum(np.power((v1-v2),2)))return dist
#加载数据
data=pd.read_excel('./电影分类数据.xlsx')
#获取训练集
train=data.iloc[:,:6]
print('train',train)
#获取训练集的特征值与目标值
train_x=train.iloc[:,:-1]
train_y=train.iloc[:,-1]
#获取测试集
test=data.columns[-4:]
print('test',test)
#进行计算距离
#循环计算每一个样本与测试集的距离
for i in range(train.shape[0]):#计算距离dist=distance(train_x.iloc[i,2:5],test[1:])train.loc[1,'dist']=dist
#对距离按照升序排序
train.sort_values(by='dist',inplace=True)
#确定K值，k值不同结果不同
k=1
res=train.loc[:,'电影类型'][:k].mode()[0]
print(res)

knn算法实现电影分类相关推荐

python实现KNN算法在电影分类中的应用
使用python实现KNN算法在电影分类中的应用代码如下: import numpy as np import operator""" 使用KNN算法对电影进行分类 K ...
KNN算法：电影分类（投票算法）
一.了解KNN K最近邻 (k-Nearest Neighbors,KNN) 算法是一种分类算法,也是最简单易懂的机器学习算法,没有之一.1968年由 Cover 和 Hart 提出,应用场景有字 ...
Knn算法实现鸢尾花分类
文章目录前言引例 KNN算法实现鸢尾花分类 1获取数据 2划分数据集 3特征工程:标准化数据 4模型训练 5模型评估总结前言近朱者赤近墨者黑,物以类聚人以群分.KNN算法就是计算和别人和自己 ...
java knn文本分类算法_使用KNN算法的文本分类.PDF
使用KNN算法的文本分类.PDF 第31 卷第8 期计算机工程 2005 年4 月 Vol.31 8 Computer Engineering April 2005 人工智能及识别技术文 ...
基于Python Scrapy爬虫改进KNN算法的网站分类系统
目录摘要 I Abstract II 第 1 章绪论 1 1.1课题的研究背景和意义 1 1.1.1目前网站分类的研究情况 1 1.1.2现有解决方案的优点与不足 1 1.1.3基于特征熵值 ...
KNN算法解决鸢尾花分类案例
KNN算法解决鸢尾花分类案例本文分别通过KNN底层算法实现和sklearn中的KNeighbors Classifier(K近邻分类模拟)和对3中不同的鸢尾花的分类. 一.K近邻(KNN)算法介绍 ...
Python编程实现用KNN算法对红酒分类功能
一.任务要求导入红酒数据集(load_wine),编写Python代码,完成以下任务: 1.实现计算平均酒精含量的功能: 2.实现对数据的标准化: 3.使用kNN算法实现红酒分类功能二.代码实现 ...
机器学习实战——KNN算法预测电影类型
预测电影类型现有爱情片和动作片(不是爱情动作片,雾)的打斗场面和接吻场面的次数统计,然后给出一个电影打斗场面和接吻场面出现的次数,预测其类型. 那么如何预测呢?当然用KNN了. KNN算法的原理就是 ...
机器学习之Python使用KNN算法进行电影类型预测
1. 使用KNN进行电影类型预测: 给定训练样本集合如下: 求解:testData={"老友记": [29, 10, 2, "?片"]}. 解题步骤: 1.计算 ...

knn算法实现电影分类

knn算法实现电影分类相关推荐

最新文章

热门文章