knn.py 文件源码

def run_knn(trainx, trainy, testx, testy):
    knn = KNeighborsClassifier(n_neighbors=5)
    knn.fit(trainx, trainy)
    pred_y = knn.predict(testx)
    print(pd.crosstab(testy, pred_y, rownames=['Actual'],
                      colnames=['Predicted']))
    print('\nAccuracy: ' + str(accuracy_score(testy, pred_y)))