KNN算法

1138-魏同学

发表文章数:75

热门标签

首页 » 算法 » 正文

1、KNN算法预测电影类别

import numpy as np
from sklearn.neighbors import KNeighborsClassifier
import pandas as pd
#分类,电影类别
#动作
#爱情
#量化,数量化,数字化
movie = pd.read_excel(r'D:/ML/4168_Sklearn_v2/4168_Sklearn_v2/V2-sklearn第一节资料/day01代码/movies.xlsx',sheet_name=1)
#sheet name 表示数据中有两个表格,加载哪一个
data = movie.iloc[:,1:3]#切片取出特定列
data
target = movie['分类情况']
display(data,target)

KNN算法

#算法学习data和target之间的关系
knn = KNeighborsClassifier(n_neighbors = 5)
#让算法进行训练
knn.fit(data,target)

KNN算法

#预测、使用算法,随便创造三个电影镜头数据
X_test = pd.DataFrame({'武打镜头':[100,67,1],'爱情':[3,2,10]})
X_test

KNN算法

#预测
knn.predict(X_test)

KNN算法

标签:

未经允许不得转载:作者:1138-魏同学, 转载或复制请以 超链接形式 并注明出处 拜师资源博客
原文地址:《KNN算法》 发布于2020-09-23

分享到:
赞(0) 打赏

评论 抢沙发

评论前必须登录!

  注册



长按图片转发给朋友

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

微信扫一扫打赏

Vieu3.3主题
专业打造轻量级个人企业风格博客主题!专注于前端开发,全站响应式布局自适应模板。

登录

忘记密码 ?

您也可以使用第三方帐号快捷登录

Q Q 登 录
微 博 登 录