机器学习之Hello World kNN

# -*- coding:utf-8 -*-
import numpy as np
import pandas as pd
from sklearn.neighbors import KNeighborsClassifier

model = KNeighborsClassifier()

origin_data = pd.read_csv('https://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data',names=['sepal_length','sepal_width','petal_length','petal_width','class_name'])

norm_data = origin_data.apply(lambda v : v if v.name == 'class_name' else (v - np.min(v))/(np.max(v) - np.min(v)))
train_data = norm_data.sample(frac=0.7)
test_data = norm_data.drop(train_data.index)

model.fit(X=train_data.iloc[:,:-1],y=np.array(train_data.iloc[:,-1:]).astype(str).ravel())
accuracy = model.score(X=test_data.iloc[:,:-1],y=np.array(test_data.iloc[:,-1:]).astype(str).ravel())
print(accuracy)
just_try = model.predict(X=test_data.iloc[:,:-1])
print(just_try)