机器学习实战——2.2预测结果不符合预期的问题解决

原创

于 2018-05-14 14:08:35 发布 · 1.3k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

在使用Python 3.6进行kNN预测时，遇到预测结果始终为'not at all'的问题。原因是test.py中的输入参数顺序与datingTestSet2.txt训练数据集顺序不一致，具体表现为飞行里程和上网时间字段顺序颠倒。修复这个问题后，程序运行正常。此外，需要注意Python 3.6中input函数替代了raw_input。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

python 3.6

我把kNN作为一个模块，另外写了一个test.py来做预测

在预测过程中发现虽然程序能跑通，但是不管怎么调整输入参数，结果大概率都是“not at all”

仔细检查后发现，是在程序输入顺序和datingTestSet2.txt中作为训练的数据集的顺序不一致，

（飞行里程和上网时间的顺序颠倒了）

修改后的程序如下：

import kNN
import numpy as np
import importlib
importlib.reload(kNN)

#约会网站预测函数
#def classifyPerson()
resultList = ['not at all','in small doses','in large doses']
ffMiles = float(input("flier miles?"))                  #注意顺序！！先飞行里程，后上网时间，最后冰淇淋
percentTats = float(input("playing video game?"))
iceCream = float(input("ice cream?"))
datingDataMat,datingLabels = kNN.file2matrix('datingTestSet2.txt')
normMat, ranges, minVals = kNN.autoNorm(datingDataMat)
inArr = np.array([ffMiles,percentTats,iceCream])         #就是这里！！我之前按错误顺序粘贴了，导致预测结果不符合实际
classifierResult = kNN.classify0((inArr-minVals)/ranges,normMat,datingLabels,3)
print ("you will like this p