九、k-近邻算法

1 简要说明

背景：各个酒店的住房记录，数据中含有row_id,x,y（x,y给出酒店位置）,accurary（酒店定位准确性）,time（签到时间）,place_id，用k-近邻算法预测进行地点预测
ps：没找到数据T-T
首先写出几点步骤：
1、需要标准化处理
2、由于数据量大，节省时间，x,y要缩小
3、时间戳转为（年，月，日，周，时分秒），当做新的特征
4、几千-几万，少于指定签到人数的位置删除

2 导入模块

from sklearn.neighbors import KNeighborsClassifier
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.preprocessing import StandardScaler

3 编写函数

def knn():
    #读取数据
    data = pd.read_csv('train.csv')
    #处理数据
    #1、缩小数据
    data.query('x>1.0 & x<1.25 & y>2.5 & y<2.75')
    #处理时间  
    time_value = pd.to_datetime(data['time'],unit='s')#最小单位为秒
    #把日期格式转为元组，好取里面的‘year’,‘month’,‘day’,‘hour’,‘second’,‘weekday’。。。
    time_value = pd.DatetimeIndex(time_value)
    #3、构造一些特征
    data['day']=time_value.day
    data['hour']=time_value.hour
    data['weekday']=time_value.weekday
    #把时间戳特征time删除
    data=data.drop(['time'],axis=1)
    #把签到数量少于n个目标位置删除
    place_count=data.groupby('place_id').count()
    tf = place_count[place_count.row_id>3].reset_index()
    data=data[data['place_id'].isin(tf.place_id)]
    #取出特征值和目标值
    y = data['place_id']
    x = data.drop(['place_id'],axis=1)
    #数据分割
    x_train,x_test,y_train,y_test = train_test_split(x,y,test_size=0.25)
    #特征工程(标准化)
    std=StandardScaler()
    #对测试集和训练集的特征值标准化
    x_train = std.fit_transform(x_train)
    x_test = std.transform(x_test)
    #进行算法流程
    knn=KNeighborsClassifier(n_neighbors=5)
    knn.fit(x_train,y_train)
    #得出预测结果
    y_predict = knn.predict(x_test)
    print('预测的目标签到位置为：',y_predict)
    #得出测试集上的准确率
    print('预测的准确率',knn.score(x_test,y_test))
if __name__=='__main__':
    knncls()

4 简单的作业

对iris数据集进行分析
首先导入数据

from sklearn.datasets import load_iris
iris = load_iris()
x = iris.data
y = iris.target

数据划分

from sklearn.model_selection import train_test_split
x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.25)

knn算法

from sklearn.neighbors import KNeighborsClassifier
from sklearn.model_selection import GridSearchCV
def knncls():
    knn = KNeighborsClassifier(n_neighbors=5)
    knn.fit(x_train,y_train)
    y_predict=knn.predict(x_test)
    print('预测结果：',y_predict)
    score = knn.score(x_test,y_test)
    print('准确性:',score)
if __name__=='__main__':
    knncls()

输出：
预测结果： [2 2 0 0 2 0 2 1 2 0 2 1 1 2 2 1 0 2 0 0 0 1 0 0 1 2 1 1 1 2 2 1 1 2 2 2 0
0]
准确性: 0.9210526315789473