赞
踩
data = pd.read_csv("./data/FBlocation/train.csv")
print(data.head())
print(data.shape)
结果:
row_id x y accuracy time place_id
0 0 0.7941 9.0809 54 470702 8523065625
1 1 5.9567 4.7968 13 186555 1757726713
2 2 8.3078 7.0407 74 322648 1137537235
3 3 7.3665 2.5165 65 704587 6567393236
4 4 4.0961 1.1307 31 472130 7440663949
(29118021, 6)
# 1、缩小数据,查询数据,为了减少计算时间 data = data.query("x > 1.0 & x < 1.25 & y > 2.5 & y < 2.75") # 2、处理时间的数据 time_value = pd.to_datetime(data['time'], unit='s') #把time转为时间类型将时间精确到秒 time_value = pd.DatetimeIndex(time_value) #设置为时间索引 data.insert(data.shape[1], 'day', time_value.day) #data.shape[1]是代表插入到最后的意思 data.insert(data.shape[1], 'hour', time_value.hour) data.insert(data.shape[1], 'weekday', time_value.weekday) data = data.drop(['time'], axis=1) # 把时间戳特征删除 # 3、把签到数量少于n个目标位置删除 place_count = data.groupby('place_id').count() #只选择去的人大于3的数据,认为1,2,3的是噪音,这个地方去的人很少,不用推荐给其他人 tf = place_count[place_count.row_id > 3].reset_index() #reset_index()重新排索引 # 根据设定的地点目标值,对原本的样本进行过滤 data = data[data['place_id'].isin(tf.place_id)] # 4、取出数据当中的特征值和目标值 y = data['place_id'] # 删除目标值,保留特征值, x = data.drop(['place_id'], axis=1) # 删除无用的特征值 x = x.drop(['row_id'], axis=1)
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.25, random_state=1)
std = StandardScaler()
# 对测试集和训练集的特征值进行标准化,分类模型的目标值不用标准化
x_train = std.fit_transform(x_train)
x_test = std.transform(x_test) #transfrom不再进行均值和方差的计算,是在原有的基础上去标准化
knn = KNeighborsClassifier()
knn.fit(x_train, y_train)
param = {"n_neighbors": [3, 5, 10, 12, 15]} #构造一些参数的值进行搜索
gc = GridSearchCV(knn, param_grid=param, cv=3) #进行网格搜索,cv=3是3折交叉验证,用其中2折训练,1折验证
gc.fit(x_train, y_train)
print("在测试集上准确率:", gc.score(x_test, y_test))
print("在交叉验证当中最好的结果:", gc.best_score_)
print("选择最好的模型是:", gc.best_estimator_)
结果:
在测试集上准确率: 0.47966903073286055
在交叉验证当中最好的结果: 0.4596467484726366
选择最好的模型是: KNeighborsClassifier(n_neighbors=10)
knn = KNeighborsClassifier(n_neighbors=10) #取参数为10
knn.fit(x_train, y_train) #训练
y_predict = knn.predict(x_test) #预测
print("预测的目标签到位置为:", y_predict)
print("预测的准确率:", knn.score(x_test, y_test))
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。