机器学习实战中如何加载kNN模块

首先，我们创建名为kNN.py的python模块。

from numpy import *
import operator

def createDataSet():
    group=array([[1.0,1.1],[1.0,1.0],[0,0],[0,0.1]])
    labels=['A','A','B','B']
    return group,labels

我们直接引用这个模块是会报错的。

In [15]: import kNN
Traceback (most recent call last):

  File "<ipython-input-15-45c55ea88398>", line 1, in <module>
    import kNN

ImportError: No module named kNN

要在python shell中调用这个函数，进入python交互开发环境

我们先使用os模块，
**查看当前路径 os.getcwd()
更改当前路径 os.chdir()**

In [17]: import os

In [19]: os.getcwd()
Out[19]: 'C:\\Users\\Administrator'

我们将kNN.py模块存放在 ‘C:\Users\Administrator’ 的路径下面。

In [20]: import kNN

In [22]: group,labels=kNN.createDataSet()

In [23]: group
Out[23]:
array([[ 1. , 1.1],
[ 1. , 1. ],
[ 0. , 0. ],
[ 0. , 0.1]])

In [24]: labels
Out[24]: ['A', 'A', 'B', 'B']

这样我们发现可以进入kNN模块，并且验证了定义的函数。