python機器學習 KNN 自己數據集

#Knn
import pandas
from sklearn.neighbors import KNeighborsClassifier
from numpy import float64
import numpy as np

#讀取數據
dataframe = pandas.read_csv("export1.csv");
#獲取 CVS中的值
dataset  = dataframe.values;
#本身數據有53列  下標我0開始 取52列  53列是標籤
X = dataset[:,0:53].astype(np.float);

Y =dataset[:,53];

#創建實例  n_neighbors K 點數
kNN_classifier = KNeighborsClassifier(n_neighbors=15);

kNN_classifier.fit(X , Y);

testData = pandas.read_csv("knnTest.csv",header=None);
#獲取 CVS中的值
testDataSet  = testData.values;
#本身數據有53列  下標我0開始 取52列  53列是標籤
xTest= testDataSet[:,0:53].astype(np.float);

returnNum =kNN_classifier.predict(xTest);

print(returnNum);