首页 > 编程语言 > 详细

k近邻算法

时间:2020-06-02 22:05:30      阅读:52      评论:0      收藏:0      [点我收藏+]
怎样计算两个数据的距离(相近程度)
假如两个数据有A B,有两个属性 p1 p2
那么,A与B的距离 distance = 开平方( 平方(A.p1 - B.p1)+ 平方(A.p2 - B.p2) )



import pandas as pd import numpy as np from sklearn import preprocessing from sklearn.neighbors import KNeighborsRegressor df = pd.read_csv("./titanic_train.csv") #drop NAN datas df = df.dropna() #75%用来训练 trainCount = int(df.shape[0]*0.75) df_train = df.iloc[:trainCount] df_test = df.iloc[trainCount:] #选择哪些列用来做比较 cols = [Age,Pclass] #初始化KNeighborsRegressor,指定取前10个最相似的近邻的平均值,默认是5个 knn = KNeighborsRegressor(n_neighbors=10) #开始训练,目标是Fare的值 knn.fit(df_train[cols],df_train[Fare]) #用测试集预测Fare值 predictions = knn.predict(df_test[cols]) print(predictions) print(df_test[Fare].values)

 

k近邻算法

原文:https://www.cnblogs.com/IWings/p/13033893.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!